Search in sources :

Example 81 with Expr

use of org.apache.druid.math.expr.Expr in project druid by druid-io.

the class TimestampExtractExprMacroTest method testApplyExtractCenturyShouldExtractTheCorrectCentury.

@Test
public void testApplyExtractCenturyShouldExtractTheCorrectCentury() {
    Expr expression = target.apply(ImmutableList.of(ExprEval.of("2000-12-16").toExpr(), ExprEval.of(TimestampExtractExprMacro.Unit.CENTURY.toString()).toExpr()));
    Assert.assertEquals(20, expression.eval(InputBindings.nilBindings()).asInt());
}
Also used : Expr(org.apache.druid.math.expr.Expr) Test(org.junit.Test)

Example 82 with Expr

use of org.apache.druid.math.expr.Expr in project druid by druid-io.

the class TimestampExtractExprMacroTest method testApplyExtractMilleniumShouldExtractTheCorrectMillenium.

@Test
public void testApplyExtractMilleniumShouldExtractTheCorrectMillenium() {
    Expr expression = target.apply(ImmutableList.of(ExprEval.of("2000-12-16").toExpr(), ExprEval.of(TimestampExtractExprMacro.Unit.MILLENNIUM.toString()).toExpr()));
    Assert.assertEquals(2, expression.eval(InputBindings.nilBindings()).asInt());
}
Also used : Expr(org.apache.druid.math.expr.Expr) Test(org.junit.Test)

Example 83 with Expr

use of org.apache.druid.math.expr.Expr in project druid by druid-io.

the class TimestampExtractExprMacroTest method testApplyExtractCenturyShouldBeTwentyFirstCenturyIn2001.

@Test
public void testApplyExtractCenturyShouldBeTwentyFirstCenturyIn2001() {
    Expr expression = target.apply(ImmutableList.of(ExprEval.of("2001-02-16").toExpr(), ExprEval.of(TimestampExtractExprMacro.Unit.CENTURY.toString()).toExpr()));
    Assert.assertEquals(21, expression.eval(InputBindings.nilBindings()).asInt());
}
Also used : Expr(org.apache.druid.math.expr.Expr) Test(org.junit.Test)

Example 84 with Expr

use of org.apache.druid.math.expr.Expr in project druid by druid-io.

the class TimestampExtractExprMacroTest method testApplyExtractMilleniumShouldBeThirdMilleniumIn2001.

@Test
public void testApplyExtractMilleniumShouldBeThirdMilleniumIn2001() {
    Expr expression = target.apply(ImmutableList.of(ExprEval.of("2001-02-16").toExpr(), ExprEval.of(TimestampExtractExprMacro.Unit.MILLENNIUM.toString()).toExpr()));
    Assert.assertEquals(3, expression.eval(InputBindings.nilBindings()).asInt());
}
Also used : Expr(org.apache.druid.math.expr.Expr) Test(org.junit.Test)

Example 85 with Expr

use of org.apache.druid.math.expr.Expr in project druid by druid-io.

the class TimestampExtractExprMacroTest method testApplyExtractDecadeShouldExtractTheCorrectDecade.

@Test
public void testApplyExtractDecadeShouldExtractTheCorrectDecade() {
    Expr expression = target.apply(ImmutableList.of(ExprEval.of("2001-02-16").toExpr(), ExprEval.of(TimestampExtractExprMacro.Unit.DECADE.toString()).toExpr()));
    Assert.assertEquals(200, expression.eval(InputBindings.nilBindings()).asInt());
}
Also used : Expr(org.apache.druid.math.expr.Expr) Test(org.junit.Test)

Aggregations

Expr (org.apache.druid.math.expr.Expr)104 Test (org.junit.Test)58 ExprEval (org.apache.druid.math.expr.ExprEval)18 InitializedNullHandlingTest (org.apache.druid.testing.InitializedNullHandlingTest)17 IAE (org.apache.druid.java.util.common.IAE)14 ExpressionType (org.apache.druid.math.expr.ExpressionType)8 DruidExpression (org.apache.druid.sql.calcite.expression.DruidExpression)7 ArrayList (java.util.ArrayList)6 Nullable (javax.annotation.Nullable)6 HashSet (java.util.HashSet)5 List (java.util.List)4 HyperLogLogCollector (org.apache.druid.hll.HyperLogLogCollector)4 BloomKFilter (org.apache.druid.query.filter.BloomKFilter)4 InDimFilter (org.apache.druid.query.filter.InDimFilter)4 RexNode (org.apache.calcite.rex.RexNode)3 Filter (org.apache.druid.query.filter.Filter)3 VirtualColumn (org.apache.druid.segment.VirtualColumn)3 FalseFilter (org.apache.druid.segment.filter.FalseFilter)3 OrFilter (org.apache.druid.segment.filter.OrFilter)3 SelectorFilter (org.apache.druid.segment.filter.SelectorFilter)3