|
9 | 9 |
|
10 | 10 | import org.apache.lucene.search.ScoreDoc; |
11 | 11 | import org.elasticsearch.test.ESTestCase; |
12 | | -import org.junit.Assert; |
13 | | -import org.junit.Test; |
14 | 12 |
|
15 | 13 | public class L2ScoreNormalizerTests extends ESTestCase { |
16 | 14 |
|
17 | | - @Test |
18 | 15 | public void testNormalizeTypicalVector() { |
19 | 16 | ScoreDoc[] docs = { new ScoreDoc(1, 3.0f, 0), new ScoreDoc(2, 4.0f, 0) }; |
20 | 17 | 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); |
23 | 20 | } |
24 | 21 |
|
25 | | - @Test |
26 | 22 | public void testAllZeros() { |
27 | 23 | ScoreDoc[] docs = { new ScoreDoc(1, 0.0f, 0), new ScoreDoc(2, 0.0f, 0) }; |
28 | 24 | 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); |
31 | 27 | } |
32 | 28 |
|
33 | | - @Test |
34 | 29 | public void testAllNaN() { |
35 | 30 | ScoreDoc[] docs = { new ScoreDoc(1, Float.NaN, 0), new ScoreDoc(2, Float.NaN, 0) }; |
36 | 31 | 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)); |
39 | 34 | } |
40 | 35 |
|
41 | | - @Test |
42 | 36 | public void testMixedZeroAndNaN() { |
43 | 37 | ScoreDoc[] docs = { new ScoreDoc(1, 0.0f, 0), new ScoreDoc(2, Float.NaN, 0) }; |
44 | 38 | 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)); |
47 | 41 | } |
48 | 42 |
|
49 | | - @Test |
50 | 43 | public void testSingleElement() { |
51 | 44 | ScoreDoc[] docs = { new ScoreDoc(1, 42.0f, 0) }; |
52 | 45 | 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); |
54 | 47 | } |
55 | 48 |
|
56 | | - @Test |
57 | 49 | public void testEmptyArray() { |
58 | 50 | ScoreDoc[] docs = {}; |
59 | 51 | ScoreDoc[] normalized = L2ScoreNormalizer.INSTANCE.normalizeScores(docs); |
60 | | - Assert.assertEquals(0, normalized.length); |
| 52 | + assertEquals(0, normalized.length); |
61 | 53 | } |
62 | 54 | } |
0 commit comments