Search in sources :

Example 1 with SubtractionTransform

use of com.linkedin.pinot.core.operator.transform.function.SubtractionTransform in project pinot by linkedin.

the class SubtractionTransformTest method test.

/**
   * Tests that the subtraction of two columns as returned by {@link SubtractionTransform
   * is as expected.
   */
@Test
public void test() {
    long seed = System.nanoTime();
    Random random = new Random(seed);
    double[] input1 = new double[NUM_ROWS];
    double[] input2 = new double[NUM_ROWS];
    double[] expected = new double[NUM_ROWS];
    for (int i = 0; i < NUM_ROWS; i++) {
        input1[i] = random.nextDouble();
        input2[i] = random.nextDouble();
        expected[i] = input1[i] - input2[i];
    }
    TransformFunction function = new SubtractionTransform();
    double[] actual = function.transform(NUM_ROWS, new TransformTestUtils.TestBlockValSet(input1, NUM_ROWS), new TransformTestUtils.TestBlockValSet(input2, NUM_ROWS));
    for (int i = 0; i < NUM_ROWS; i++) {
        Assert.assertEquals(actual[i], expected[i], EPSILON, ("Random seed: " + seed));
    }
}
Also used : SubtractionTransform(com.linkedin.pinot.core.operator.transform.function.SubtractionTransform) Random(java.util.Random) TransformFunction(com.linkedin.pinot.core.operator.transform.function.TransformFunction) Test(org.testng.annotations.Test)

Aggregations

SubtractionTransform (com.linkedin.pinot.core.operator.transform.function.SubtractionTransform)1 TransformFunction (com.linkedin.pinot.core.operator.transform.function.TransformFunction)1 Random (java.util.Random)1 Test (org.testng.annotations.Test)1