diff options
author | Jon Bratseth <bratseth@vespa.ai> | 2024-02-26 14:27:27 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@vespa.ai> | 2024-02-26 14:27:27 +0100 |
commit | 8a32b84caaa9219a67210c389fd643154ef10ec6 (patch) | |
tree | 0a0f2fbf1afecc963d7bee50e810387b11323f63 /container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java | |
parent | 1f4fe9ac7726c7501cf4414f9fc009a8564096ce (diff) |
Resolve alias references
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java')
-rw-r--r-- | container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java b/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java index 8e7c7276de1..7bbc1801204 100644 --- a/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java +++ b/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java @@ -223,19 +223,33 @@ public class ValidateNearestNeighborTestCase { @Test void testWithQueryProfileArgument() { + testWithQueryProfileArgument("foo"); + } + + @Test + void testWithQueryProfileArgumentFromBuiltInProperties() { + testWithQueryProfileArgument("model.queryTree"); + } + + @Test + void testWithQueryProfileArgumentFromBuiltInPropertyAlias() { + testWithQueryProfileArgument("query"); + } + + private void testWithQueryProfileArgument(String argument) { var embedder = new MockEmbedder("test text", Language.UNKNOWN, Tensor.from("tensor<float>(x[3]):[1.0, 2.0, 3.0]")); var registry = new QueryProfileRegistry(); var profile = new QueryProfile("test"); - profile.set("ranking.features.query(qvector)", "embed(@foo)", registry); + profile.set("ranking.features.query(qvector)", "embed(@" + argument + ")", registry); registry.register(profile); var queryString = makeQuery("fvector", "qvector"); var query = new Query.Builder() .setSchemaInfo(createSchemaInfo()) .setQueryProfile(registry.compile().findQueryProfile("test")) .setEmbedder(embedder) - .setRequestMap(Map.of("foo", "test text")) + .setRequestMap(Map.of(argument, "test text")) .build(); setYqlQuery(query, queryString); var result = doSearch(searcher, query); |