|
7 | 7 |
|
8 | 8 | package org.elasticsearch.xpack.rank.linear; |
9 | 9 |
|
10 | | -import org.apache.lucene.search.MatchAllDocsQuery; |
11 | 10 | import org.apache.lucene.search.TotalHits; |
12 | 11 | import org.elasticsearch.ElasticsearchStatusException; |
13 | 12 | import org.elasticsearch.ExceptionsHelper; |
|
37 | 36 | import org.elasticsearch.search.collapse.CollapseBuilder; |
38 | 37 | import org.elasticsearch.search.retriever.CompoundRetrieverBuilder; |
39 | 38 | import org.elasticsearch.search.retriever.KnnRetrieverBuilder; |
40 | | -import org.elasticsearch.search.retriever.RetrieverBuilder; |
41 | 39 | import org.elasticsearch.search.retriever.StandardRetrieverBuilder; |
42 | 40 | import org.elasticsearch.search.retriever.TestRetrieverBuilder; |
43 | 41 | import org.elasticsearch.search.sort.FieldSortBuilder; |
|
60 | 58 | import java.util.List; |
61 | 59 | import java.util.concurrent.TimeUnit; |
62 | 60 | import java.util.concurrent.atomic.AtomicInteger; |
63 | | -import java.util.Map; |
64 | 61 |
|
65 | 62 | import static org.elasticsearch.cluster.metadata.IndexMetadata.SETTING_NUMBER_OF_SHARDS; |
66 | 63 | import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertResponse; |
@@ -905,7 +902,10 @@ public void testLinearWithMinScore() { |
905 | 902 | ), |
906 | 903 | rankWindowSize, |
907 | 904 | new float[] { 1.0f, 1.0f, 0.0f }, |
908 | | - new ScoreNormalizer[] { IdentityScoreNormalizer.INSTANCE, IdentityScoreNormalizer.INSTANCE, IdentityScoreNormalizer.INSTANCE }, |
| 905 | + new ScoreNormalizer[] { |
| 906 | + IdentityScoreNormalizer.INSTANCE, |
| 907 | + IdentityScoreNormalizer.INSTANCE, |
| 908 | + IdentityScoreNormalizer.INSTANCE }, |
909 | 909 | 10.0f |
910 | 910 | ) |
911 | 911 | ); |
@@ -995,11 +995,11 @@ public void testLinearWithMinScoreAndNormalization() { |
995 | 995 |
|
996 | 996 | // Calculated scores >= 1.5f should only be doc_2(2.0) |
997 | 997 | // Observed behavior consistently shows 2 hits: doc_2(2.0) and one other doc (doc_1 or doc_3) with score 0.0 |
998 | | - assertThat(resp.getHits().getHits().length, equalTo(2)); |
| 998 | + assertThat(resp.getHits().getHits().length, equalTo(2)); |
999 | 999 | assertThat(resp.getHits().getAt(0).getId(), equalTo("doc_2")); |
1000 | 1000 | assertThat((double) resp.getHits().getAt(0).getScore(), closeTo(2.0f, 0.01f)); |
1001 | 1001 | // Assert the second hit has score 0.0, but don't assert its ID due to inconsistency |
1002 | | - assertThat((double) resp.getHits().getAt(1).getScore(), closeTo(0.0f, 0.01f)); |
| 1002 | + assertThat((double) resp.getHits().getAt(1).getScore(), closeTo(0.0f, 0.01f)); |
1003 | 1003 | }); |
1004 | 1004 | } |
1005 | 1005 |
|
|
0 commit comments