Skip to content

Commit a014e02

Browse files
committed
Updated LinearRetrieverBuilderParsingTests
1 parent e8715b2 commit a014e02

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

x-pack/plugin/rank-rrf/src/test/java/org/elasticsearch/xpack/rank/linear/LinearRetrieverBuilderParsingTests.java

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,16 @@ public static void afterClass() throws Exception {
4343
@Override
4444
protected LinearRetrieverBuilder createTestInstance() {
4545
int rankWindowSize = randomInt(100);
46+
47+
List<String> fields = null;
48+
String query = null;
49+
ScoreNormalizer normalizer = null;
50+
if (randomBoolean()) {
51+
fields = randomList(1, 10, () -> randomAlphaOfLengthBetween(1, 10));
52+
query = randomAlphaOfLengthBetween(1, 10);
53+
normalizer = randomScoreNormalizer();
54+
}
55+
4656
int num = randomIntBetween(1, 3);
4757
List<CompoundRetrieverBuilder.RetrieverSource> innerRetrievers = new ArrayList<>();
4858
float[] weights = new float[num];
@@ -54,7 +64,8 @@ protected LinearRetrieverBuilder createTestInstance() {
5464
weights[i] = randomFloat();
5565
normalizers[i] = randomScoreNormalizer();
5666
}
57-
return new LinearRetrieverBuilder(innerRetrievers, rankWindowSize, weights, normalizers);
67+
68+
return new LinearRetrieverBuilder(innerRetrievers, fields, query, normalizer, rankWindowSize, weights, normalizers);
5869
}
5970

6071
@Override

0 commit comments

Comments
 (0)