Skip to content

Commit a0ab7e3

Browse files
committed
Fixed the forbidden test
1 parent 2ccb6ac commit a0ab7e3

File tree

1 file changed

+10
-18
lines changed

1 file changed

+10
-18
lines changed

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

Lines changed: 10 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -9,54 +9,46 @@
99

1010
import org.apache.lucene.search.ScoreDoc;
1111
import org.elasticsearch.test.ESTestCase;
12-
import org.junit.Assert;
13-
import org.junit.Test;
1412

1513
public class L2ScoreNormalizerTests extends ESTestCase {
1614

17-
@Test
1815
public void testNormalizeTypicalVector() {
1916
ScoreDoc[] docs = { new ScoreDoc(1, 3.0f, 0), new ScoreDoc(2, 4.0f, 0) };
2017
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
21-
Assert.assertEquals(0.6f, normalized[0].score, 1e-5);
22-
Assert.assertEquals(0.8f, normalized[1].score, 1e-5);
18+
assertEquals(0.6f, normalized[0].score, 1e-5);
19+
assertEquals(0.8f, normalized[1].score, 1e-5);
2320
}
2421

25-
@Test
2622
public void testAllZeros() {
2723
ScoreDoc[] docs = { new ScoreDoc(1, 0.0f, 0), new ScoreDoc(2, 0.0f, 0) };
2824
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
29-
Assert.assertEquals(0.0f, normalized[0].score, 0.0f);
30-
Assert.assertEquals(0.0f, normalized[1].score, 0.0f);
25+
assertEquals(0.0f, normalized[0].score, 0.0f);
26+
assertEquals(0.0f, normalized[1].score, 0.0f);
3127
}
3228

33-
@Test
3429
public void testAllNaN() {
3530
ScoreDoc[] docs = { new ScoreDoc(1, Float.NaN, 0), new ScoreDoc(2, Float.NaN, 0) };
3631
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
37-
Assert.assertTrue(Float.isNaN(normalized[0].score));
38-
Assert.assertTrue(Float.isNaN(normalized[1].score));
32+
assertTrue(Float.isNaN(normalized[0].score));
33+
assertTrue(Float.isNaN(normalized[1].score));
3934
}
4035

41-
@Test
4236
public void testMixedZeroAndNaN() {
4337
ScoreDoc[] docs = { new ScoreDoc(1, 0.0f, 0), new ScoreDoc(2, Float.NaN, 0) };
4438
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
45-
Assert.assertEquals(0.0f, normalized[0].score, 0.0f);
46-
Assert.assertTrue(Float.isNaN(normalized[1].score));
39+
assertEquals(0.0f, normalized[0].score, 0.0f);
40+
assertTrue(Float.isNaN(normalized[1].score));
4741
}
4842

49-
@Test
5043
public void testSingleElement() {
5144
ScoreDoc[] docs = { new ScoreDoc(1, 42.0f, 0) };
5245
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
53-
Assert.assertEquals(1.0f, normalized[0].score, 1e-5);
46+
assertEquals(1.0f, normalized[0].score, 1e-5);
5447
}
5548

56-
@Test
5749
public void testEmptyArray() {
5850
ScoreDoc[] docs = {};
5951
ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs);
60-
Assert.assertEquals(0, normalized.length);
52+
assertEquals(0, normalized.length);
6153
}
6254
}

0 commit comments

Comments
 (0)