use of org.apache.druid.segment.BaseLongColumnValueSelector in project druid by druid-io.
the class ExpressionVirtualColumnTest method testLongSelectorUsingStringFunction.
@Test
public void testLongSelectorUsingStringFunction() {
final BaseLongColumnValueSelector selector = Z_CONCAT_X.makeColumnValueSelector("expr", COLUMN_SELECTOR_FACTORY);
CURRENT_ROW.set(ROW0);
if (NullHandling.replaceWithDefault()) {
Assert.assertEquals(0L, selector.getLong());
} else {
Assert.assertTrue(selector.isNull());
}
CURRENT_ROW.set(ROW1);
if (NullHandling.replaceWithDefault()) {
Assert.assertEquals(4L, selector.getLong());
} else {
// y is null for row1
Assert.assertTrue(selector.isNull());
}
CURRENT_ROW.set(ROW2);
if (NullHandling.replaceWithDefault()) {
Assert.assertEquals(0L, selector.getLong());
} else {
Assert.assertTrue(selector.isNull());
}
CURRENT_ROW.set(ROW3);
if (NullHandling.replaceWithDefault()) {
Assert.assertEquals(0L, selector.getLong());
} else {
Assert.assertTrue(selector.isNull());
}
}
use of org.apache.druid.segment.BaseLongColumnValueSelector in project druid by druid-io.
the class ExpressionVirtualColumnTest method testLongSelectorWithConstantNullArithmetic.
@Test
public void testLongSelectorWithConstantNullArithmetic() {
final BaseLongColumnValueSelector selector = CONSTANT_NULL_ARITHMETIC.makeColumnValueSelector("expr", COLUMN_SELECTOR_FACTORY);
CURRENT_ROW.set(ROW0);
if (NullHandling.replaceWithDefault()) {
Assert.assertEquals(2L, selector.getLong());
Assert.assertFalse(selector.isNull());
} else {
Assert.assertTrue(selector.isNull());
}
}
use of org.apache.druid.segment.BaseLongColumnValueSelector in project druid by druid-io.
the class VirtualColumnsTest method testMakeSelectorsWithDotSupportBaseNameOnly.
@Test
public void testMakeSelectorsWithDotSupportBaseNameOnly() {
final VirtualColumns virtualColumns = makeVirtualColumns();
final BaseObjectColumnValueSelector objectSelector = virtualColumns.makeColumnValueSelector("foo", null);
final DimensionSelector dimensionSelector = virtualColumns.makeDimensionSelector(new DefaultDimensionSpec("foo", "x"), null);
final BaseFloatColumnValueSelector floatSelector = virtualColumns.makeColumnValueSelector("foo", null);
final BaseLongColumnValueSelector longSelector = virtualColumns.makeColumnValueSelector("foo", null);
Assert.assertEquals(-1L, objectSelector.getObject());
Assert.assertEquals("-1", dimensionSelector.lookupName(dimensionSelector.getRow().get(0)));
Assert.assertEquals(-1.0f, floatSelector.getFloat(), 0.0f);
Assert.assertEquals(-1L, longSelector.getLong());
}
Aggregations