aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search/significance
diff options
context:
space:
mode:
authorMariusArhaug <mariusarhaug@hotmail.com>2024-04-24 13:28:41 +0200
committerMariusArhaug <mariusarhaug@hotmail.com>2024-04-30 14:42:32 +0200
commitad8bfd90c688d39315485b3748fc2d27e38daf7d (patch)
treef5f39714b005c2a1ebaf037bb4d5c65b00ce7e1a /container-search/src/test/java/com/yahoo/search/significance
parent0a20132a7a507d15cb51a51e25d2a8081bb6c94f (diff)
Add significance rank profile
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search/significance')
-rw-r--r--container-search/src/test/java/com/yahoo/search/significance/test/SignificanceSearcherTest.java18
1 files changed, 17 insertions, 1 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 ad39c49bc1b..ed67798b4b1 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
@@ -2,6 +2,7 @@
package com.yahoo.search.significance.test;
import com.yahoo.component.chain.Chain;
+import com.yahoo.config.subscription.ConfigGetter;
import com.yahoo.language.Language;
import com.yahoo.language.significance.SignificanceModel;
import com.yahoo.language.significance.SignificanceModelRegistry;
@@ -12,6 +13,7 @@ import com.yahoo.search.Query;
import com.yahoo.search.Result;
import com.yahoo.search.searchchain.Execution;
import com.yahoo.search.significance.SignificanceSearcher;
+import com.yahoo.vespa.config.search.RankProfilesConfig;
import org.junit.jupiter.api.Test;
import java.nio.file.Path;
@@ -31,12 +33,24 @@ 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());
+
significanceModelRegistry = new DefaultSignificanceModelRegistry(models);
- searcher = new SignificanceSearcher(significanceModelRegistry);
+ 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);
}
private Execution createExecution(SignificanceSearcher searcher) {
@@ -51,6 +65,7 @@ public class SignificanceSearcherTest {
void testSignificanceValueOnSimpleQuery() {
Query q = new Query();
+ q.getRanking().setProfile("significance-ranking");
AndItem root = new AndItem();
WordItem tmp;
tmp = new WordItem("Hello", true);
@@ -81,6 +96,7 @@ public class SignificanceSearcherTest {
@Test
void testSignificanceValueOnRecursiveQuery() {
Query q = new Query();
+ q.getRanking().setProfile("significance-ranking");
AndItem root = new AndItem();
WordItem child1 = new WordItem("hello", true);