From 437bc6d89b13e1ac745d5e0ccfb47415d8b8bd2a Mon Sep 17 00:00:00 2001 From: Arne Juul Date: Tue, 3 Oct 2023 11:46:18 +0000 Subject: add unit test --- .../yahoo/search/ranking/NormalizerTestCase.java | 64 ++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 container-search/src/test/java/com/yahoo/search/ranking/NormalizerTestCase.java (limited to 'container-search/src/test/java/com/yahoo/search') diff --git a/container-search/src/test/java/com/yahoo/search/ranking/NormalizerTestCase.java b/container-search/src/test/java/com/yahoo/search/ranking/NormalizerTestCase.java new file mode 100644 index 00000000000..1c10ad083a3 --- /dev/null +++ b/container-search/src/test/java/com/yahoo/search/ranking/NormalizerTestCase.java @@ -0,0 +1,64 @@ +// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.search.ranking; + +import org.junit.jupiter.api.Test; + +import static org.junit.jupiter.api.Assertions.assertEquals; + +/** + * @author arnej + */ +public class NormalizerTestCase { + + @Test + void requireLinearNormalizing() { + var n = new LinearNormalizer("foo", "bar", 10); + assertEquals(0, n.addInput(-4.0)); + assertEquals(1, n.addInput(-1.0)); + assertEquals(2, n.addInput(-5.0)); + assertEquals(3, n.addInput(-3.0)); + n.normalize(); + assertEquals(0.0, n.getOutput(2)); + assertEquals(0.25, n.getOutput(0)); + assertEquals(0.5, n.getOutput(3)); + assertEquals(1.0, n.getOutput(1)); + assertEquals("foo", n.name()); + assertEquals("bar", n.input()); + assertEquals("linear", n.normalizing()); + } + + @Test + void requireReciprocalNormalizing() { + var n = new ReciprocalRankNormalizer("foo", "bar", 10, 0.0); + assertEquals(0, n.addInput(-4.1)); + assertEquals(1, n.addInput(11.0)); + assertEquals(2, n.addInput(-50.0)); + assertEquals(3, n.addInput(-3.0)); + n.normalize(); + assertEquals(0.25, n.getOutput(2)); + assertEquals(0.3333333, n.getOutput(0), 0.00001); + assertEquals(0.5, n.getOutput(3)); + assertEquals(1.0, n.getOutput(1)); + assertEquals("foo", n.name()); + assertEquals("bar", n.input()); + assertEquals("reciprocal-rank{k:0.0}", n.normalizing()); + } + + @Test + void requireReciprocalNormalizingWithK() { + var n = new ReciprocalRankNormalizer("foo", "bar", 10, 4.2); + assertEquals(0, n.addInput(-4.1)); + assertEquals(1, n.addInput(11.0)); + assertEquals(2, n.addInput(-50.0)); + assertEquals(3, n.addInput(-3.0)); + n.normalize(); + assertEquals(1.0/8.2, n.getOutput(2)); + assertEquals(1.0/7.2, n.getOutput(0), 0.00001); + assertEquals(1.0/6.2, n.getOutput(3)); + assertEquals(1.0/5.2, n.getOutput(1)); + assertEquals("foo", n.name()); + assertEquals("bar", n.input()); + assertEquals("reciprocal-rank{k:4.2}", n.normalizing()); + } + +} -- cgit v1.2.3