aboutsummaryrefslogtreecommitdiff
path: root/icing/scoring/scoring-processor_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'icing/scoring/scoring-processor_test.cc')
-rw-r--r--icing/scoring/scoring-processor_test.cc63
1 files changed, 33 insertions, 30 deletions
diff --git a/icing/scoring/scoring-processor_test.cc b/icing/scoring/scoring-processor_test.cc
index b42ba31..9ca7dfd 100644
--- a/icing/scoring/scoring-processor_test.cc
+++ b/icing/scoring/scoring-processor_test.cc
@@ -329,19 +329,19 @@ TEST_F(ScoringProcessorTest,
DocumentId document_id3,
document_store()->Put(document3, /*num_tokens=*/50));
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
doc_hit_info2.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
- DocHitInfo doc_hit_info3(document_id3);
+ DocHitInfoTermFrequencyPair doc_hit_info3 = DocHitInfo(document_id3);
doc_hit_info3.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
SectionId section_id = 0;
- SectionIdMask section_id_mask = 1U << section_id;
+ SectionIdMask section_id_mask = UINT64_C(1) << section_id;
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2,
- doc_hit_info3};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {
+ doc_hit_info1, doc_hit_info2, doc_hit_info3};
// Creates a dummy DocHitInfoIterator with 3 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -398,19 +398,19 @@ TEST_F(ScoringProcessorTest,
DocumentId document_id3,
document_store()->Put(document3, /*num_tokens=*/10));
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
doc_hit_info2.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
- DocHitInfo doc_hit_info3(document_id3);
+ DocHitInfoTermFrequencyPair doc_hit_info3 = DocHitInfo(document_id3);
doc_hit_info3.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
SectionId section_id = 0;
- SectionIdMask section_id_mask = 1U << section_id;
+ SectionIdMask section_id_mask = UINT64_C(1) << section_id;
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2,
- doc_hit_info3};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {
+ doc_hit_info1, doc_hit_info2, doc_hit_info3};
// Creates a dummy DocHitInfoIterator with 3 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -466,13 +466,13 @@ TEST_F(ScoringProcessorTest,
DocumentId document_id3,
document_store()->Put(document3, /*num_tokens=*/10));
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
// Document 1 contains the query term "foo" 5 times
doc_hit_info1.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/5);
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
// Document 1 contains the query term "foo" 1 time
doc_hit_info2.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
- DocHitInfo doc_hit_info3(document_id3);
+ DocHitInfoTermFrequencyPair doc_hit_info3 = DocHitInfo(document_id3);
// Document 1 contains the query term "foo" 3 times
doc_hit_info3.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/1);
doc_hit_info3.UpdateSection(/*section_id*/ 1, /*hit_term_frequency=*/2);
@@ -482,8 +482,8 @@ TEST_F(ScoringProcessorTest,
SectionIdMask section_id_mask3 = 0b00000011;
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2,
- doc_hit_info3};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {
+ doc_hit_info1, doc_hit_info2, doc_hit_info3};
// Creates a dummy DocHitInfoIterator with 3 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -528,11 +528,11 @@ TEST_F(ScoringProcessorTest,
document_store()->Put(document1, /*num_tokens=*/10));
// Document 1 contains the term "foo" 0 times in the "subject" property
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(/*section_id*/ 0, /*hit_term_frequency=*/0);
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {doc_hit_info1};
// Creates a dummy DocHitInfoIterator with 1 result for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -580,16 +580,17 @@ TEST_F(ScoringProcessorTest,
// Document 1 contains the term "foo" 1 time in the "body" property
SectionId body_section_id = 0;
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(body_section_id, /*hit_term_frequency=*/1);
// Document 2 contains the term "foo" 1 time in the "subject" property
SectionId subject_section_id = 1;
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
doc_hit_info2.UpdateSection(subject_section_id, /*hit_term_frequency=*/1);
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {doc_hit_info1,
+ doc_hit_info2};
// Creates a dummy DocHitInfoIterator with 2 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -651,16 +652,17 @@ TEST_F(ScoringProcessorTest,
// Document 1 contains the term "foo" 1 time in the "body" property
SectionId body_section_id = 0;
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(body_section_id, /*hit_term_frequency=*/1);
// Document 2 contains the term "foo" 1 time in the "subject" property
SectionId subject_section_id = 1;
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
doc_hit_info2.UpdateSection(subject_section_id, /*hit_term_frequency=*/1);
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {doc_hit_info1,
+ doc_hit_info2};
// Creates a dummy DocHitInfoIterator with 2 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -719,11 +721,11 @@ TEST_F(ScoringProcessorTest,
// Document 1 contains the term "foo" 1 time in the "body" property
SectionId body_section_id = 0;
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(body_section_id, /*hit_term_frequency=*/1);
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {doc_hit_info1};
// Creates a dummy DocHitInfoIterator with 1 result for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
@@ -809,16 +811,17 @@ TEST_F(ScoringProcessorTest,
// Document 1 contains the term "foo" 1 time in the "body" property
SectionId body_section_id = 0;
- DocHitInfo doc_hit_info1(document_id1);
+ DocHitInfoTermFrequencyPair doc_hit_info1 = DocHitInfo(document_id1);
doc_hit_info1.UpdateSection(body_section_id, /*hit_term_frequency=*/1);
// Document 2 contains the term "foo" 1 time in the "subject" property
SectionId subject_section_id = 1;
- DocHitInfo doc_hit_info2(document_id2);
+ DocHitInfoTermFrequencyPair doc_hit_info2 = DocHitInfo(document_id2);
doc_hit_info2.UpdateSection(subject_section_id, /*hit_term_frequency=*/1);
// Creates input doc_hit_infos and expected output scored_document_hits
- std::vector<DocHitInfo> doc_hit_infos = {doc_hit_info1, doc_hit_info2};
+ std::vector<DocHitInfoTermFrequencyPair> doc_hit_infos = {doc_hit_info1,
+ doc_hit_info2};
// Creates a dummy DocHitInfoIterator with 2 results for the query "foo"
std::unique_ptr<DocHitInfoIterator> doc_hit_info_iterator =
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-13 04:44:50 +0000