aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
authorGeir Storli <geirst@yahooinc.com>2022-07-08 12:25:02 +0000
committerGeir Storli <geirst@yahooinc.com>2022-07-08 12:26:17 +0000
commitf6897716bee6f446fe5047ca4fbc8aafff27138e (patch)
tree2155aabd2120695e283f7b3c0a4245e5ae1f066a /searchlib
parentebf6eda095d61b1a1c5b755b98062d39a5c58f0b (diff)
Share fixture classes used by distance and closeness unit tests.
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/CMakeLists.txt1
-rw-r--r--searchlib/src/tests/features/nns_closeness/CMakeLists.txt1
-rw-r--r--searchlib/src/tests/features/nns_closeness/nns_closeness_test.cpp102
-rw-r--r--searchlib/src/tests/features/nns_distance/CMakeLists.txt1
-rw-r--r--searchlib/src/tests/features/nns_distance/nns_distance_test.cpp106
-rw-r--r--searchlib/src/vespa/searchlib/fef/test/labels.h2
-rw-r--r--searchlib/src/vespa/searchlib/test/CMakeLists.txt1
-rw-r--r--searchlib/src/vespa/searchlib/test/features/CMakeLists.txt6
-rw-r--r--searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.cpp10
-rw-r--r--searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.h102
10 files changed, 146 insertions, 186 deletions
diff --git a/searchlib/CMakeLists.txt b/searchlib/CMakeLists.txt
index 43a8da19191..f9b6271b47b 100644
--- a/searchlib/CMakeLists.txt
+++ b/searchlib/CMakeLists.txt
@@ -46,6 +46,7 @@ vespa_define_module(
src/vespa/searchlib/test
src/vespa/searchlib/test/diskindex
src/vespa/searchlib/test/fakedata
+ src/vespa/searchlib/test/features
src/vespa/searchlib/test/memoryindex
src/vespa/searchlib/transactionlog
src/vespa/searchlib/uca
diff --git a/searchlib/src/tests/features/nns_closeness/CMakeLists.txt b/searchlib/src/tests/features/nns_closeness/CMakeLists.txt
index 9a8c2d7c99f..d5f9ece096b 100644
--- a/searchlib/src/tests/features/nns_closeness/CMakeLists.txt
+++ b/searchlib/src/tests/features/nns_closeness/CMakeLists.txt
@@ -5,5 +5,6 @@ vespa_add_executable(searchlib_nns_closeness_test_app TEST
nns_closeness_test.cpp
DEPENDS
searchlib
+ searchlib_test
)
vespa_add_test(NAME searchlib_nns_closeness_test_app COMMAND searchlib_nns_closeness_test_app)
diff --git a/searchlib/src/tests/features/nns_closeness/nns_closeness_test.cpp b/searchlib/src/tests/features/nns_closeness/nns_closeness_test.cpp
index c7667b2cecd..661ee884e46 100644
--- a/searchlib/src/tests/features/nns_closeness/nns_closeness_test.cpp
+++ b/searchlib/src/tests/features/nns_closeness/nns_closeness_test.cpp
@@ -1,106 +1,24 @@
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
-#include <vespa/vespalib/testkit/test_kit.h>
-#include <vespa/searchlib/features/setup.h>
-#include <vespa/searchlib/fef/test/indexenvironment.h>
-#include <vespa/searchlib/fef/test/indexenvironmentbuilder.h>
-#include <vespa/searchlib/fef/test/queryenvironment.h>
-#include <vespa/searchlib/fef/test/labels.h>
#include <vespa/searchlib/features/closenessfeature.h>
-#include <vespa/searchlib/fef/fef.h>
+#include <vespa/searchlib/features/setup.h>
#include <vespa/searchlib/fef/test/dummy_dependency_handler.h>
+#include <vespa/searchlib/fef/test/labels.h>
+#include <vespa/searchlib/test/features/distance_closeness_fixture.h>
#include <vespa/vespalib/stllike/asciistream.h>
+#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/vespalib/util/stringfmt.h>
using search::feature_t;
-using namespace search::fef;
-using namespace search::fef::test;
+using namespace search::features::test;
using namespace search::features;
-using CollectionType = FieldInfo::CollectionType;
-using DataType = FieldInfo::DataType;
+using namespace search::fef::test;
+using namespace search::fef;
const vespalib::string labelFeatureName("closeness(label,nns)");
const vespalib::string fieldFeatureName("closeness(bar)");
-struct BlueprintFactoryFixture {
- BlueprintFactory factory;
- BlueprintFactoryFixture() : factory()
- {
- setup_search_features(factory);
- }
-};
-
-struct IndexFixture {
- IndexEnvironment indexEnv;
- IndexFixture() : indexEnv()
- {
- IndexEnvironmentBuilder builder(indexEnv);
- builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::INT64, "foo");
- builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::TENSOR, "bar");
- }
-};
-
-struct FeatureDumpFixture : public IDumpFeatureVisitor {
- virtual void visitDumpFeature(const vespalib::string &) override {
- TEST_ERROR("no features should be dumped");
- }
- FeatureDumpFixture() : IDumpFeatureVisitor() {}
- ~FeatureDumpFixture() override;
-};
-
-FeatureDumpFixture::~FeatureDumpFixture() = default;
-
-struct RankFixture : BlueprintFactoryFixture, IndexFixture {
- QueryEnvironment queryEnv;
- RankSetup rankSetup;
- MatchDataLayout mdl;
- MatchData::UP match_data;
- RankProgram::UP rankProgram;
- std::vector<TermFieldHandle> fooHandles;
- std::vector<TermFieldHandle> barHandles;
- RankFixture(size_t fooCnt, size_t barCnt, const Labels &labels, const vespalib::string &featureName)
- : queryEnv(&indexEnv), rankSetup(factory, indexEnv),
- mdl(), match_data(), rankProgram(), fooHandles(), barHandles()
- {
- for (size_t i = 0; i < fooCnt; ++i) {
- uint32_t fieldId = indexEnv.getFieldByName("foo")->id();
- fooHandles.push_back(mdl.allocTermField(fieldId));
- SimpleTermData term;
- term.setUniqueId(i + 1);
- term.addField(fieldId).setHandle(fooHandles.back());
- queryEnv.getTerms().push_back(term);
- }
- for (size_t i = 0; i < barCnt; ++i) {
- uint32_t fieldId = indexEnv.getFieldByName("bar")->id();
- barHandles.push_back(mdl.allocTermField(fieldId));
- SimpleTermData term;
- term.setUniqueId(fooCnt + i + 1);
- term.addField(fieldId).setHandle(barHandles.back());
- queryEnv.getTerms().push_back(term);
- }
- labels.inject(queryEnv.getProperties());
- rankSetup.setFirstPhaseRank(featureName);
- rankSetup.setIgnoreDefaultRankFeatures(true);
- ASSERT_TRUE(rankSetup.compile());
- match_data = mdl.createMatchData();
- rankProgram = rankSetup.create_first_phase_program();
- rankProgram->setup(*match_data, queryEnv);
- }
- feature_t getScore(uint32_t docId) {
- return Utils::getScoreFeature(*rankProgram, docId);
- }
- void setScore(TermFieldHandle handle, uint32_t docId, feature_t score) {
- match_data->resolveTermField(handle)->setRawScore(docId, score);
- }
- void setFooScore(uint32_t i, uint32_t docId, feature_t distance) {
- ASSERT_LESS(i, fooHandles.size());
- setScore(fooHandles[i], docId, 1.0/(1.0+distance));
- }
- void setBarScore(uint32_t i, uint32_t docId, feature_t distance) {
- ASSERT_LESS(i, barHandles.size());
- setScore(barHandles[i], docId, 1.0/(1.0+distance));
- }
-};
+using RankFixture = DistanceClosenessFixture;
TEST_F("require that blueprint can be created from factory", BlueprintFactoryFixture) {
Blueprint::SP bp = f.factory.createBlueprint("closeness");
@@ -108,11 +26,11 @@ TEST_F("require that blueprint can be created from factory", BlueprintFactoryFix
EXPECT_TRUE(dynamic_cast<ClosenessBlueprint*>(bp.get()) != 0);
}
-TEST_FFF("require that no features are dumped", ClosenessBlueprint, IndexFixture, FeatureDumpFixture) {
+TEST_FFF("require that no features are dumped", ClosenessBlueprint, IndexEnvironmentFixture, FeatureDumpFixture) {
f1.visitDumpFeatures(f2.indexEnv, f3);
}
-TEST_FF("require that setup can be done on random label", ClosenessBlueprint, IndexFixture) {
+TEST_FF("require that setup can be done on random label", ClosenessBlueprint, IndexEnvironmentFixture) {
DummyDependencyHandler deps(f1);
f1.setName(vespalib::make_string("%s(label,random_label)", f1.getBaseName().c_str()));
EXPECT_TRUE(static_cast<Blueprint&>(f1).setup(f2.indexEnv, std::vector<vespalib::string>{"label", "random_label"}));
diff --git a/searchlib/src/tests/features/nns_distance/CMakeLists.txt b/searchlib/src/tests/features/nns_distance/CMakeLists.txt
index 8dbee37a194..bf4e533ea45 100644
--- a/searchlib/src/tests/features/nns_distance/CMakeLists.txt
+++ b/searchlib/src/tests/features/nns_distance/CMakeLists.txt
@@ -5,5 +5,6 @@ vespa_add_executable(searchlib_nns_distance_test_app TEST
nns_distance_test.cpp
DEPENDS
searchlib
+ searchlib_test
)
vespa_add_test(NAME searchlib_nns_distance_test_app COMMAND searchlib_nns_distance_test_app)
diff --git a/searchlib/src/tests/features/nns_distance/nns_distance_test.cpp b/searchlib/src/tests/features/nns_distance/nns_distance_test.cpp
index 1e81b5576c1..6b2669367ad 100644
--- a/searchlib/src/tests/features/nns_distance/nns_distance_test.cpp
+++ b/searchlib/src/tests/features/nns_distance/nns_distance_test.cpp
@@ -2,105 +2,23 @@
#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/searchlib/features/setup.h>
-#include <vespa/searchlib/fef/test/indexenvironment.h>
-#include <vespa/searchlib/fef/test/indexenvironmentbuilder.h>
-#include <vespa/searchlib/fef/test/queryenvironment.h>
#include <vespa/searchlib/fef/test/labels.h>
#include <vespa/searchlib/features/distancefeature.h>
-#include <vespa/searchlib/fef/fef.h>
#include <vespa/searchlib/fef/test/dummy_dependency_handler.h>
+#include <vespa/searchlib/test/features/distance_closeness_fixture.h>
#include <vespa/vespalib/stllike/asciistream.h>
#include <vespa/vespalib/util/stringfmt.h>
using search::feature_t;
-using namespace search::fef;
-using namespace search::fef::test;
+using namespace search::features::test;
using namespace search::features;
-using CollectionType = FieldInfo::CollectionType;
-using DataType = FieldInfo::DataType;
+using namespace search::fef::test;
+using namespace search::fef;
-const vespalib::string labelFeatureName("distance(label,label)");
+const vespalib::string labelFeatureName("distance(label,nns)");
const vespalib::string fieldFeatureName("distance(bar)");
-struct BlueprintFactoryFixture {
- BlueprintFactory factory;
- BlueprintFactoryFixture() : factory()
- {
- setup_search_features(factory);
- }
-};
-
-struct IndexFixture {
- IndexEnvironment indexEnv;
- IndexFixture() : indexEnv()
- {
- IndexEnvironmentBuilder builder(indexEnv);
- builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::INT64, "foo");
- builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::TENSOR, "bar");
- }
-};
-
-struct FeatureDumpFixture : public IDumpFeatureVisitor {
- virtual void visitDumpFeature(const vespalib::string &) override {
- TEST_ERROR("no features should be dumped");
- }
- FeatureDumpFixture() : IDumpFeatureVisitor() {}
- ~FeatureDumpFixture() override;
-};
-
-FeatureDumpFixture::~FeatureDumpFixture() = default;
-
-struct RankFixture : BlueprintFactoryFixture, IndexFixture {
- QueryEnvironment queryEnv;
- RankSetup rankSetup;
- MatchDataLayout mdl;
- MatchData::UP match_data;
- RankProgram::UP rankProgram;
- std::vector<TermFieldHandle> fooHandles;
- std::vector<TermFieldHandle> barHandles;
- RankFixture(size_t fooCnt, size_t barCnt, const Labels &labels, const vespalib::string &featureName)
- : queryEnv(&indexEnv), rankSetup(factory, indexEnv),
- mdl(), match_data(), rankProgram(), fooHandles(), barHandles()
- {
- for (size_t i = 0; i < fooCnt; ++i) {
- uint32_t fieldId = indexEnv.getFieldByName("foo")->id();
- fooHandles.push_back(mdl.allocTermField(fieldId));
- SimpleTermData term;
- term.setUniqueId(i + 1);
- term.addField(fieldId).setHandle(fooHandles.back());
- queryEnv.getTerms().push_back(term);
- }
- for (size_t i = 0; i < barCnt; ++i) {
- uint32_t fieldId = indexEnv.getFieldByName("bar")->id();
- barHandles.push_back(mdl.allocTermField(fieldId));
- SimpleTermData term;
- term.setUniqueId(fooCnt + i + 1);
- term.addField(fieldId).setHandle(barHandles.back());
- queryEnv.getTerms().push_back(term);
- }
- labels.inject(queryEnv.getProperties());
- rankSetup.setFirstPhaseRank(featureName);
- rankSetup.setIgnoreDefaultRankFeatures(true);
- ASSERT_TRUE(rankSetup.compile());
- match_data = mdl.createMatchData();
- rankProgram = rankSetup.create_first_phase_program();
- rankProgram->setup(*match_data, queryEnv);
- }
- feature_t getScore(uint32_t docId) {
- return Utils::getScoreFeature(*rankProgram, docId);
- }
- void setScore(TermFieldHandle handle, uint32_t docId, feature_t score) {
- match_data->resolveTermField(handle)->setRawScore(docId, score);
- }
- void setFooScore(uint32_t i, uint32_t docId, feature_t distance) {
- ASSERT_LESS(i, fooHandles.size());
- setScore(fooHandles[i], docId, 1.0/(1.0+distance));
- }
- void setBarScore(uint32_t i, uint32_t docId, feature_t distance) {
- ASSERT_LESS(i, barHandles.size());
- setScore(barHandles[i], docId, 1.0/(1.0+distance));
- }
-};
+using RankFixture = DistanceClosenessFixture;
TEST_F("require that blueprint can be created from factory", BlueprintFactoryFixture) {
Blueprint::SP bp = f.factory.createBlueprint("distance");
@@ -108,17 +26,17 @@ TEST_F("require that blueprint can be created from factory", BlueprintFactoryFix
EXPECT_TRUE(dynamic_cast<DistanceBlueprint*>(bp.get()) != 0);
}
-TEST_FFF("require that no features are dumped", DistanceBlueprint, IndexFixture, FeatureDumpFixture) {
+TEST_FFF("require that no features are dumped", DistanceBlueprint, IndexEnvironmentFixture, FeatureDumpFixture) {
f1.visitDumpFeatures(f2.indexEnv, f3);
}
-TEST_FF("require that setup can be done on random label", DistanceBlueprint, IndexFixture) {
+TEST_FF("require that setup can be done on random label", DistanceBlueprint, IndexEnvironmentFixture) {
DummyDependencyHandler deps(f1);
f1.setName(vespalib::make_string("%s(label,random_label)", f1.getBaseName().c_str()));
EXPECT_TRUE(static_cast<Blueprint&>(f1).setup(f2.indexEnv, std::vector<vespalib::string>{"label", "random_label"}));
}
-TEST_FF("require that setup with unknown field fails", DistanceBlueprint, IndexFixture) {
+TEST_FF("require that setup with unknown field fails", DistanceBlueprint, IndexEnvironmentFixture) {
DummyDependencyHandler deps(f1);
f1.setName(vespalib::make_string("%s(field,random_fieldname)", f1.getBaseName().c_str()));
EXPECT_FALSE(static_cast<Blueprint&>(f1).setup(f2.indexEnv, std::vector<vespalib::string>{"field", "random_fieldname"}));
@@ -132,7 +50,7 @@ TEST_FF("require that unrelated label gives max-double distance", SingleLabel("u
EXPECT_EQUAL(std::numeric_limits<feature_t>::max(), f2.getScore(10));
}
-TEST_FF("require that labeled item raw score can be obtained", SingleLabel("label", 1), RankFixture(2, 2, f1, labelFeatureName)) {
+TEST_FF("require that labeled item raw score can be obtained", SingleLabel("nns", 1), RankFixture(2, 2, f1, labelFeatureName)) {
f2.setFooScore(0, 10, 5.0);
EXPECT_EQUAL(5.0, f2.getScore(10));
}
@@ -142,7 +60,7 @@ TEST_FF("require that field raw score can be obtained", NoLabel(), RankFixture(2
EXPECT_EQUAL(5.0, f2.getScore(10));
}
-TEST_FF("require that other raw scores are ignored", SingleLabel("label", 2), RankFixture(2, 2, f1, labelFeatureName)) {
+TEST_FF("require that other raw scores are ignored", SingleLabel("nns", 2), RankFixture(2, 2, f1, labelFeatureName)) {
f2.setFooScore(0, 10, 1.0);
f2.setFooScore(1, 10, 2.0);
f2.setBarScore(0, 10, 5.0);
@@ -158,7 +76,7 @@ TEST_FF("require that the correct raw score is used", NoLabel(), RankFixture(2,
EXPECT_EQUAL(7.0, f2.getScore(10));
}
-TEST_FF("require that stale data is ignored", SingleLabel("label", 2), RankFixture(2, 2, f1, labelFeatureName)) {
+TEST_FF("require that stale data is ignored", SingleLabel("nns", 2), RankFixture(2, 2, f1, labelFeatureName)) {
f2.setFooScore(0, 10, 1.0);
f2.setFooScore(1, 5, 2.0);
EXPECT_EQUAL(std::numeric_limits<feature_t>::max(), f2.getScore(10));
diff --git a/searchlib/src/vespa/searchlib/fef/test/labels.h b/searchlib/src/vespa/searchlib/fef/test/labels.h
index 5f27f786405..a69634003b4 100644
--- a/searchlib/src/vespa/searchlib/fef/test/labels.h
+++ b/searchlib/src/vespa/searchlib/fef/test/labels.h
@@ -1,5 +1,7 @@
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#pragma once
+
#include <vespa/searchlib/fef/properties.h>
#include <vespa/vespalib/stllike/asciistream.h>
diff --git a/searchlib/src/vespa/searchlib/test/CMakeLists.txt b/searchlib/src/vespa/searchlib/test/CMakeLists.txt
index 0d56fb8b4be..ed884a46217 100644
--- a/searchlib/src/vespa/searchlib/test/CMakeLists.txt
+++ b/searchlib/src/vespa/searchlib/test/CMakeLists.txt
@@ -13,6 +13,7 @@ vespa_add_library(searchlib_test
$<TARGET_OBJECTS:searchlib_test_gtest_migration>
DEPENDS
searchlib
+ searchlib_searchlib_test_features
searchlib_searchlib_test_memoryindex
GTest::GTest
)
diff --git a/searchlib/src/vespa/searchlib/test/features/CMakeLists.txt b/searchlib/src/vespa/searchlib/test/features/CMakeLists.txt
new file mode 100644
index 00000000000..00b7073bf03
--- /dev/null
+++ b/searchlib/src/vespa/searchlib/test/features/CMakeLists.txt
@@ -0,0 +1,6 @@
+# Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+vespa_add_library(searchlib_searchlib_test_features
+ SOURCES
+ distance_closeness_fixture.cpp
+ DEPENDS
+)
diff --git a/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.cpp b/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.cpp
new file mode 100644
index 00000000000..6ec81c8dbf0
--- /dev/null
+++ b/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.cpp
@@ -0,0 +1,10 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include "distance_closeness_fixture.h"
+
+namespace search::features::test {
+
+FeatureDumpFixture::~FeatureDumpFixture() = default;
+
+}
+
diff --git a/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.h b/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.h
new file mode 100644
index 00000000000..6063c22763b
--- /dev/null
+++ b/searchlib/src/vespa/searchlib/test/features/distance_closeness_fixture.h
@@ -0,0 +1,102 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#pragma once
+
+#include <vespa/searchlib/features/setup.h>
+#include <vespa/searchlib/fef/fef.h>
+#include <vespa/searchlib/fef/test/indexenvironment.h>
+#include <vespa/searchlib/fef/test/indexenvironmentbuilder.h>
+#include <vespa/searchlib/fef/test/labels.h>
+#include <vespa/searchlib/fef/test/queryenvironment.h>
+#include <vespa/vespalib/testkit/test_kit.h>
+
+using namespace search::fef;
+using namespace search::fef::test;
+
+using CollectionType = FieldInfo::CollectionType;
+using DataType = FieldInfo::DataType;
+
+namespace search::features::test {
+
+struct BlueprintFactoryFixture {
+ BlueprintFactory factory;
+ BlueprintFactoryFixture() : factory()
+ {
+ setup_search_features(factory);
+ }
+};
+
+struct IndexEnvironmentFixture {
+ IndexEnvironment indexEnv;
+ IndexEnvironmentFixture() : indexEnv()
+ {
+ IndexEnvironmentBuilder builder(indexEnv);
+ builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::INT64, "foo");
+ builder.addField(FieldType::ATTRIBUTE, CollectionType::SINGLE, DataType::TENSOR, "bar");
+ }
+};
+
+struct FeatureDumpFixture : public IDumpFeatureVisitor {
+ virtual void visitDumpFeature(const vespalib::string &) override {
+ TEST_ERROR("no features should be dumped");
+ }
+ FeatureDumpFixture() : IDumpFeatureVisitor() {}
+ ~FeatureDumpFixture() override;
+};
+
+/**
+ * Fixture used by unit tests for distance and closeness rank features.
+ */
+struct DistanceClosenessFixture : BlueprintFactoryFixture, IndexEnvironmentFixture {
+ QueryEnvironment queryEnv;
+ RankSetup rankSetup;
+ MatchDataLayout mdl;
+ MatchData::UP match_data;
+ RankProgram::UP rankProgram;
+ std::vector<TermFieldHandle> fooHandles;
+ std::vector<TermFieldHandle> barHandles;
+ DistanceClosenessFixture(size_t fooCnt, size_t barCnt, const Labels &labels, const vespalib::string &featureName)
+ : queryEnv(&indexEnv), rankSetup(factory, indexEnv),
+ mdl(), match_data(), rankProgram(), fooHandles(), barHandles()
+ {
+ for (size_t i = 0; i < fooCnt; ++i) {
+ uint32_t fieldId = indexEnv.getFieldByName("foo")->id();
+ fooHandles.push_back(mdl.allocTermField(fieldId));
+ SimpleTermData term;
+ term.setUniqueId(i + 1);
+ term.addField(fieldId).setHandle(fooHandles.back());
+ queryEnv.getTerms().push_back(term);
+ }
+ for (size_t i = 0; i < barCnt; ++i) {
+ uint32_t fieldId = indexEnv.getFieldByName("bar")->id();
+ barHandles.push_back(mdl.allocTermField(fieldId));
+ SimpleTermData term;
+ term.setUniqueId(fooCnt + i + 1);
+ term.addField(fieldId).setHandle(barHandles.back());
+ queryEnv.getTerms().push_back(term);
+ }
+ labels.inject(queryEnv.getProperties());
+ rankSetup.setFirstPhaseRank(featureName);
+ rankSetup.setIgnoreDefaultRankFeatures(true);
+ ASSERT_TRUE(rankSetup.compile());
+ match_data = mdl.createMatchData();
+ rankProgram = rankSetup.create_first_phase_program();
+ rankProgram->setup(*match_data, queryEnv);
+ }
+ feature_t getScore(uint32_t docId) {
+ return Utils::getScoreFeature(*rankProgram, docId);
+ }
+ void setScore(TermFieldHandle handle, uint32_t docId, feature_t score) {
+ match_data->resolveTermField(handle)->setRawScore(docId, score);
+ }
+ void setFooScore(uint32_t i, uint32_t docId, feature_t distance) {
+ ASSERT_LESS(i, fooHandles.size());
+ setScore(fooHandles[i], docId, 1.0/(1.0+distance));
+ }
+ void setBarScore(uint32_t i, uint32_t docId, feature_t distance) {
+ ASSERT_LESS(i, barHandles.size());
+ setScore(barHandles[i], docId, 1.0/(1.0+distance));
+ }
+};
+
+}