diff options
author | Geir Storli <geirst@yahooinc.com> | 2023-04-25 19:30:56 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-25 19:30:56 +0200 |
commit | 51f5ab6eb26d901e5018094338db7cd59ded9106 (patch) | |
tree | b962417a8cb38c23fcc0d4166fd6c3a23fc17fd7 /streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp | |
parent | b552ce33c561eef8b4440bb2ddc93b24afb8d16a (diff) | |
parent | 1d6fcfb5c5b7399cd33c32ffea30fd9208ec000b (diff) |
Merge pull request #26850 from vespa-engine/geirst/nearest-neighbor-target-hits-in-streaming
Use targetHits in nearestNeighbor streaming searcher.
Diffstat (limited to 'streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp')
-rw-r--r-- | streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp b/streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp index 9f3f3d770e4..4d425d9dedd 100644 --- a/streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp +++ b/streamingvisitors/src/tests/rank_processor/rank_processor_test.cpp @@ -55,6 +55,10 @@ RankProcessorTest::build_query(QueryBuilder<SimpleQueryNodeTypes> &builder) _query_wrapper = std::make_unique<QueryWrapper>(*_query); } +class MockRawScoreCalculator : public search::streaming::NearestNeighborQueryNode::RawScoreCalculator { +public: + double to_raw_score(double distance) override { return distance * 2; } +}; TEST_F(RankProcessorTest, unpack_match_data_for_nearest_neighbor_query_node) { @@ -71,6 +75,8 @@ TEST_F(RankProcessorTest, unpack_match_data_for_nearest_neighbor_query_node) EXPECT_EQ(1u, term_list.size()); auto node = dynamic_cast<NearestNeighborQueryNode*>(term_list.front().getTerm()); EXPECT_NE(nullptr, node); + MockRawScoreCalculator calc; + node->set_raw_score_calc(&calc); auto& qtd = static_cast<QueryTermData &>(node->getQueryItem()); auto& td = qtd.getTermData(); constexpr TermFieldHandle handle = 27; @@ -82,11 +88,11 @@ TEST_F(RankProcessorTest, unpack_match_data_for_nearest_neighbor_query_node) EXPECT_EQ(invalid_id, tfmd->getDocId()); RankProcessor::unpack_match_data(1, *md, *_query_wrapper); EXPECT_EQ(invalid_id, tfmd->getDocId()); - constexpr double raw_score = 1.5; - node->set_raw_score(raw_score); + constexpr double distance = 1.5; + node->set_distance(distance); RankProcessor::unpack_match_data(2, *md, *_query_wrapper); EXPECT_EQ(2, tfmd->getDocId()); - EXPECT_EQ(raw_score, tfmd->getRawScore()); + EXPECT_EQ(distance * 2, tfmd->getRawScore()); node->reset(); RankProcessor::unpack_match_data(3, *md, *_query_wrapper); EXPECT_EQ(2, tfmd->getDocId()); |