aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@vespa.ai>2024-05-06 11:25:02 +0200
committerBjørn Christian Seime <bjorncs@vespa.ai>2024-05-06 11:25:02 +0200
commit633782b31fed623d6d2f2a48d8bebbc5b9840be0 (patch)
treed8d129174c76a5a9bc038c11dec103ea9b5a7ad8 /container-search/src/test/java/com/yahoo/search
parent2f4511677d4da29e615f3543fd167d4bbce8588e (diff)
Configure signficance searcher through `SchemaInfo`
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search')
-rw-r--r--container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java22
1 files changed, 10 insertions, 12 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java b/container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java
index ed67798b4b1..abda1df473f 100644
--- a/container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java
+++ b/container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java
@@ -11,6 +11,10 @@ import com.yahoo.prelude.query.AndItem;
import com.yahoo.prelude.query.WordItem;
import com.yahoo.search.Query;
import com.yahoo.search.Result;
+import com.yahoo.search.schema.DocumentSummary;
+import com.yahoo.search.schema.RankProfile;
+import com.yahoo.search.schema.Schema;
+import com.yahoo.search.schema.SchemaInfo;
import com.yahoo.search.searchchain.Execution;
import com.yahoo.search.significance.SignificanceSearcher;
import com.yahoo.vespa.config.search.RankProfilesConfig;
@@ -33,24 +37,18 @@ public class SignificanceSearcherTest {
SignificanceModelRegistry significanceModelRegistry;
SignificanceSearcher searcher;
- private static final String CONFIG_DIR = "src/test/resources/config/";
public SignificanceSearcherTest() {
List<Path> models = new ArrayList<>();
models.add( Path.of("src/test/java/com/yahoo/search/significance/model/en.json"));
- RankProfilesConfig rpCfg = readConfig("with_significance");
-
- assertEquals(1, rpCfg.rankprofile().size());
-
+ var schema = new Schema.Builder("music")
+ .add(new DocumentSummary.Builder("default").build())
+ .add(new RankProfile.Builder("significance-ranking")
+ .setUseSignificanceModel(true)
+ .build());
significanceModelRegistry = new DefaultSignificanceModelRegistry(models);
- searcher = new SignificanceSearcher(significanceModelRegistry, rpCfg);
- }
-
- @SuppressWarnings("deprecation")
- private RankProfilesConfig readConfig(String subDir) {
- String cfgId = "file:" + CONFIG_DIR + subDir + "/rank-profiles.cfg";
- return ConfigGetter.getConfig(RankProfilesConfig.class, cfgId);
+ searcher = new SignificanceSearcher(significanceModelRegistry, new SchemaInfo(List.of(schema.build()), List.of()));
}
private Execution createExecution(SignificanceSearcher searcher) {