Search in sources :

Example 46 with ExpressionResult

use of org.apache.carbondata.core.scan.expression.ExpressionResult in project carbondata by apache.

the class EqualToExpressionUnitTest method testEvaluateForEqualToExpressionWithShortDataType.

@Test
public void testEvaluateForEqualToExpressionWithShortDataType() throws FilterUnsupportedException, FilterIllegalMemberException {
    ColumnExpression right = new ColumnExpression("id", DataType.SHORT);
    right.setColIndex(0);
    equalToExpression = new EqualToExpression(right, right);
    RowImpl value = new RowImpl();
    Short[] row = { 15 };
    Object[] objectRow = { row };
    value.setValues(objectRow);
    new MockUp<ExpressionResult>() {

        @Mock
        public Short getShort() {
            return 15;
        }
    };
    ExpressionResult result = equalToExpression.evaluate(value);
    assertTrue(result.getBoolean());
}
Also used : RowImpl(org.apache.carbondata.core.scan.filter.intf.RowImpl) ExpressionResult(org.apache.carbondata.core.scan.expression.ExpressionResult) ColumnExpression(org.apache.carbondata.core.scan.expression.ColumnExpression) MockUp(mockit.MockUp) Test(org.junit.Test)

Example 47 with ExpressionResult

use of org.apache.carbondata.core.scan.expression.ExpressionResult in project carbondata by apache.

the class GreaterThanEqualToExpressionUnitTest method testEvaluateForGreaterThanEqualToExpressionWithIsNullReturnTrue.

@Test
public void testEvaluateForGreaterThanEqualToExpressionWithIsNullReturnTrue() throws FilterUnsupportedException, FilterIllegalMemberException {
    ColumnExpression left = new ColumnExpression("id", DataType.SHORT);
    left.setColIndex(0);
    ColumnExpression right = new ColumnExpression("id", DataType.SHORT);
    right.setColIndex(1);
    greaterThanEqualToExpression = new GreaterThanEqualToExpression(left, right);
    RowImpl value = new RowImpl();
    Short[] row = { 15 };
    Object[] objectRow = { row, row };
    value.setValues(objectRow);
    new MockUp<ExpressionResult>() {

        @Mock
        public boolean isNull() {
            return true;
        }
    };
    new MockUp<ExpressionResult>() {

        @Mock
        public Short getShort() {
            return 15;
        }
    };
    ExpressionResult result = greaterThanEqualToExpression.evaluate(value);
    assertFalse(result.getBoolean());
}
Also used : RowImpl(org.apache.carbondata.core.scan.filter.intf.RowImpl) ExpressionResult(org.apache.carbondata.core.scan.expression.ExpressionResult) ColumnExpression(org.apache.carbondata.core.scan.expression.ColumnExpression) MockUp(mockit.MockUp) Test(org.junit.Test)

Example 48 with ExpressionResult

use of org.apache.carbondata.core.scan.expression.ExpressionResult in project carbondata by apache.

the class GreaterThanExpressionUnitTest method testEvaluateForGreaterThanExpressionWithIntDataType.

@Test
public void testEvaluateForGreaterThanExpressionWithIntDataType() throws FilterUnsupportedException, FilterIllegalMemberException {
    ColumnExpression right = new ColumnExpression("right_number", DataType.INT);
    right.setColIndex(0);
    ColumnExpression left = new ColumnExpression("left_number", DataType.INT);
    left.setColIndex(1);
    greaterThanExpression = new GreaterThanExpression(left, right);
    RowImpl value = new RowImpl();
    Integer[] row = { 140 };
    Integer[] row1 = { 150 };
    Object[] objectRow = { row, row1 };
    value.setValues(objectRow);
    new MockUp<ExpressionResult>() {

        Boolean returnMockFlag = true;

        @Mock
        public Integer getInt() {
            if (returnMockFlag) {
                returnMockFlag = false;
                return 150;
            } else {
                return 140;
            }
        }
    };
    ExpressionResult result = greaterThanExpression.evaluate(value);
    assertTrue(result.getBoolean());
}
Also used : RowImpl(org.apache.carbondata.core.scan.filter.intf.RowImpl) ExpressionResult(org.apache.carbondata.core.scan.expression.ExpressionResult) ColumnExpression(org.apache.carbondata.core.scan.expression.ColumnExpression) MockUp(mockit.MockUp) Test(org.junit.Test)

Example 49 with ExpressionResult

use of org.apache.carbondata.core.scan.expression.ExpressionResult in project carbondata by apache.

the class GreaterThanExpressionUnitTest method testEvaluateForGreaterThanExpressionWithLongDataType.

@Test
public void testEvaluateForGreaterThanExpressionWithLongDataType() throws FilterUnsupportedException, FilterIllegalMemberException {
    ColumnExpression right = new ColumnExpression("contact", DataType.LONG);
    right.setColIndex(0);
    ColumnExpression left = new ColumnExpression("contact", DataType.LONG);
    left.setColIndex(1);
    greaterThanExpression = new GreaterThanExpression(left, right);
    RowImpl value = new RowImpl();
    Long[] row = { 1234567654321L };
    Long[] row1 = { 123456765432234L };
    Object[] objectRow = { row1, row };
    value.setValues(objectRow);
    new MockUp<ExpressionResult>() {

        Boolean returnMockFlag = true;

        @Mock
        public Long getLong() {
            if (returnMockFlag) {
                returnMockFlag = false;
                return 123456765432234L;
            } else {
                return 1234567654321L;
            }
        }
    };
    ExpressionResult result = greaterThanExpression.evaluate(value);
    assertTrue(result.getBoolean());
}
Also used : RowImpl(org.apache.carbondata.core.scan.filter.intf.RowImpl) ExpressionResult(org.apache.carbondata.core.scan.expression.ExpressionResult) ColumnExpression(org.apache.carbondata.core.scan.expression.ColumnExpression) MockUp(mockit.MockUp) Test(org.junit.Test)

Example 50 with ExpressionResult

use of org.apache.carbondata.core.scan.expression.ExpressionResult in project carbondata by apache.

the class GreaterThanExpressionUnitTest method testEvaluateForGreaterThanExpressionWithTimestampDataType.

@Test
public void testEvaluateForGreaterThanExpressionWithTimestampDataType() throws FilterUnsupportedException, FilterIllegalMemberException {
    try {
        ColumnExpression left = new ColumnExpression("timestamp", DataType.TIMESTAMP);
        left.setColIndex(0);
        ColumnExpression right = new ColumnExpression("timestamp", DataType.TIMESTAMP);
        right.setColIndex(1);
        greaterThanExpression = new GreaterThanExpression(left, right);
        RowImpl value = new RowImpl();
        DateFormat dateFormat = new SimpleDateFormat("dd/MM/yyyy");
        Date date = dateFormat.parse("23/09/2007");
        long time = date.getTime();
        Timestamp[] row = { new Timestamp(time) };
        Date date1 = dateFormat.parse("24/09/2007");
        long time1 = date1.getTime();
        Timestamp[] row1 = { new Timestamp(time1) };
        Object[] objectRow = { row1, row };
        value.setValues(objectRow);
        new MockUp<ExpressionResult>() {

            Boolean returnMockFlag = true;

            @Mock
            public Long getTime() {
                if (returnMockFlag) {
                    returnMockFlag = false;
                    return 1190592000L;
                } else {
                    return 1190505600L;
                }
            }
        };
        ExpressionResult result = greaterThanExpression.evaluate(value);
        assertTrue(result.getBoolean());
    } catch (ParseException e) {
        System.out.println("Error while parsing " + e.getMessage());
    }
}
Also used : RowImpl(org.apache.carbondata.core.scan.filter.intf.RowImpl) ExpressionResult(org.apache.carbondata.core.scan.expression.ExpressionResult) MockUp(mockit.MockUp) Timestamp(java.sql.Timestamp) Date(java.util.Date) ColumnExpression(org.apache.carbondata.core.scan.expression.ColumnExpression) SimpleDateFormat(java.text.SimpleDateFormat) DateFormat(java.text.DateFormat) ParseException(java.text.ParseException) SimpleDateFormat(java.text.SimpleDateFormat) Test(org.junit.Test)

Aggregations

ExpressionResult (org.apache.carbondata.core.scan.expression.ExpressionResult)99 RowImpl (org.apache.carbondata.core.scan.filter.intf.RowImpl)81 Test (org.junit.Test)81 MockUp (mockit.MockUp)78 ColumnExpression (org.apache.carbondata.core.scan.expression.ColumnExpression)77 FilterUnsupportedException (org.apache.carbondata.core.scan.expression.exception.FilterUnsupportedException)14 BigDecimal (java.math.BigDecimal)8 Timestamp (java.sql.Timestamp)8 DateFormat (java.text.DateFormat)8 ParseException (java.text.ParseException)8 SimpleDateFormat (java.text.SimpleDateFormat)8 Date (java.util.Date)8 Decimal (org.apache.spark.sql.types.Decimal)8 ArrayList (java.util.ArrayList)6 FilterIllegalMemberException (org.apache.carbondata.core.scan.expression.exception.FilterIllegalMemberException)5 RowIntf (org.apache.carbondata.core.scan.filter.intf.RowIntf)4 Expression (org.apache.carbondata.core.scan.expression.Expression)3 BinaryConditionalExpression (org.apache.carbondata.core.scan.expression.conditional.BinaryConditionalExpression)3 LiteralExpression (org.apache.carbondata.core.scan.expression.LiteralExpression)2 DimColumnFilterInfo (org.apache.carbondata.core.scan.filter.DimColumnFilterInfo)2