| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.tajo.engine.eval; |
| |
| import org.apache.tajo.catalog.CatalogUtil; |
| import org.apache.tajo.catalog.Schema; |
| import org.apache.tajo.catalog.SchemaFactory; |
| import org.apache.tajo.common.TajoDataTypes; |
| import org.apache.tajo.exception.TajoException; |
| import org.junit.Test; |
| |
| import static org.apache.tajo.common.TajoDataTypes.Type.BOOLEAN; |
| import static org.apache.tajo.common.TajoDataTypes.Type.INT4; |
| import static org.apache.tajo.common.TajoDataTypes.Type.TEXT; |
| |
| public class TestPredicates extends ExprTestBase { |
| |
| ////////////////////////////////////////////////////////////////// |
| // Logical Operator |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testAnd() throws TajoException { |
| testSimpleEval("select true;", new String[] {"t"}); |
| |
| testSimpleEval("select true and true;", new String[] {"t"}); |
| testSimpleEval("select true and false;", new String[] {"f"}); |
| testSimpleEval("select false and true;", new String[] {"f"}); |
| testSimpleEval("select false and false;", new String[] {"f"}); |
| } |
| |
| @Test |
| public void testOr() throws TajoException { |
| testSimpleEval("select true or true;", new String[] {"t"}); |
| testSimpleEval("select true or false;", new String[] {"t"}); |
| testSimpleEval("select false or true;", new String[] {"t"}); |
| testSimpleEval("select false or false;", new String[] {"f"}); |
| } |
| |
| @Test |
| public void testLogicalOperatorPrecedence() throws TajoException { |
| testSimpleEval("select true or (false or false) or false;", new String[] {"t"}); |
| testSimpleEval("select false or (true or false) or false;", new String[] {"t"}); |
| testSimpleEval("select false or (false or true) or false;", new String[] {"t"}); |
| testSimpleEval("select false or (false or false) or true;", new String[] {"t"}); |
| |
| testSimpleEval("select true and (false or false) or false;", new String[] {"f"}); |
| testSimpleEval("select false and (true or false) or false;", new String[] {"f"}); |
| testSimpleEval("select false and (false or true) or false;", new String[] {"f"}); |
| testSimpleEval("select false and (false or false) or true;", new String[] {"t"}); |
| |
| testSimpleEval("select true or (false and false) or false;", new String[] {"t"}); |
| testSimpleEval("select false or (true and false) or false;", new String[] {"f"}); |
| testSimpleEval("select false or (false and true) or false;", new String[] {"f"}); |
| testSimpleEval("select false or (false and true) or true;", new String[] {"t"}); |
| |
| testSimpleEval("select true or (false or false) and false;", new String[] {"t"}); |
| testSimpleEval("select false or (true or false) and false;", new String[] {"f"}); |
| testSimpleEval("select false or (false or true) and false;", new String[] {"f"}); |
| testSimpleEval("select false or (false or false) and true;", new String[] {"f"}); |
| } |
| |
| @Test |
| public void testNot() throws TajoException { |
| |
| testSimpleEval("select true;", new String[] {"t"}); |
| testSimpleEval("select not true;", new String[] {"f"}); |
| testSimpleEval("select (true);", new String[] {"t"}); |
| testSimpleEval("select not (true);", new String[] {"f"}); |
| testSimpleEval("select not (not (true));", new String[] {"t"}); |
| |
| testSimpleEval("select (not (1 > null)) is null;", new String[] {"t"}); |
| |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 = col2, col1 = col3 from table1", |
| new String[]{"t", "f"}); |
| } |
| |
| @Test |
| public void testParenthesizedValues() throws TajoException { |
| testSimpleEval("select ((true));", new String[] {"t"}); |
| testSimpleEval("select ((((true))));", new String[] {"t"}); |
| testSimpleEval("select not(not(not(false)));", new String[] {"t"}); |
| } |
| |
| ////////////////////////////////////////////////////////////////// |
| // Comparison Predicate |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testComparisonEqual() throws TajoException { |
| |
| |
| Schema schema = SchemaFactory.newV1(); |
| schema.addColumn("col0", TajoDataTypes.Type.INT1); |
| schema.addColumn("col1", TajoDataTypes.Type.INT2); |
| schema.addColumn("col2", TajoDataTypes.Type.INT4); |
| schema.addColumn("col3", TajoDataTypes.Type.INT8); |
| schema.addColumn("col4", TajoDataTypes.Type.FLOAT4); |
| schema.addColumn("col5", TajoDataTypes.Type.FLOAT8); |
| schema.addColumn("col6", TajoDataTypes.Type.TEXT); |
| schema.addColumn("col7", CatalogUtil.newDataType(TajoDataTypes.Type.CHAR, "", 3)); |
| schema.addColumn("nullable", TajoDataTypes.Type.INT4); |
| |
| testEval(schema, "t1", "0,1,2,3,4.1,5.1,cmp,asm,", "SELECT col6 = 'cmp' from t1", new String [] {"t"}); |
| |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 = col2, col1 = col3, col1 = col4 from table1", |
| new String[]{"t", "f", "f"}); |
| testEval(schema1, |
| "table1", "123,123,,", |
| "select col1 = col2, (col1 = col3) is null, (col3 = col2) is null from table1", |
| new String[]{"t", "t", "t"}); |
| } |
| |
| @Test |
| public void testComparisonNotEqual() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 <> col2, col1 <> col3, col1 <> col4 from table1", |
| new String[]{"f", "t", "t"}); |
| testEval(schema1, |
| "table1", "123,123,,", |
| "select col1 <> col2, (col1 <> col3) is null, (col3 <> col2) is null from table1", |
| new String[]{"f", "t", "t"}); |
| } |
| |
| @Test |
| public void testComparisonLessThan() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 < col2, col1 < col3, col1 < col4 from table1", |
| new String[]{"f", "t", "f"}); |
| testEval(schema1, |
| "table1", "123,456,,", |
| "select col1 < col2, (col1 = col3) is null, (col4 = col1) is null from table1", |
| new String[]{"t", "t", "t"}); |
| } |
| |
| @Test |
| public void testComparisonLessThanEqual() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 <= col2, col1 <= col3, col1 <= col4 from table1", |
| new String[]{"t", "t", "f"}); |
| testEval(schema1, |
| "table1", "123,456,,", |
| "select col1 <= col2, (col1 <= col3) is null, (col4 <= col1) is null from table1", |
| new String[]{"t", "t", "t"}); |
| } |
| |
| @Test |
| public void testComparisonGreaterThan() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 > col2, col3 > col2, col1 > col4 from table1", |
| new String[]{"f", "t", "t"}); |
| testEval(schema1, |
| "table1", "123,456,,", |
| "select col2 > col1, col1 > col2, (col1 > col3) is null, (col4 > col1) is null from table1", |
| new String[]{"t", "f", "t", "t"}); |
| } |
| |
| @Test |
| public void testComparisonGreaterThanEqual() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| schema1.addColumn("col3", INT4); |
| schema1.addColumn("col4", INT4); |
| testEval(schema1, |
| "table1", "123,123,456,-123", |
| "select col1 >= col2, col3 >= col2, col1 >= col4 from table1", |
| new String[]{"t", "t", "t"}); |
| testEval(schema1, |
| "table1", "123,456,,", |
| "select col2 >= col1, col1 >= col2, (col1 >= col3) is null, (col4 >= col1) is null from table1", |
| new String[]{"t", "f", "t", "t"}); |
| } |
| |
| ////////////////////////////////////////////////////////////////// |
| // Between Predicate |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testBetween() throws TajoException { |
| Schema schema2 = SchemaFactory.newV1(); |
| schema2.addColumn("col1", TEXT); |
| schema2.addColumn("col2", TEXT); |
| schema2.addColumn("col3", TEXT); |
| |
| // constant checker |
| testEval(schema2, "table1", "b,a,c", "select col1 between 'a' and 'c' from table1", new String[]{"t"}); |
| testEval(schema2, "table1", "b,a,c", "select col1 between 'c' and 'a' from table1", new String[]{"f"}); |
| testEval(schema2, "table1", "b,a,c", "select col1 between symmetric 'c' and 'a' from table1", new String[]{"t"}); |
| testEval(schema2, "table1", "d,a,c", "select col1 between 'a' and 'c' from table1", new String[]{"f"}); |
| |
| // tests for inclusive |
| testEval(schema2, "table1", "a,a,c", "select col1 between col2 and col3 from table1", new String[]{"t"}); |
| testEval(schema2, "table1", "b,a,c", "select col1 between col2 and col3 from table1", new String[]{"t"}); |
| testEval(schema2, "table1", "c,a,c", "select col1 between col2 and col3 from table1", new String[]{"t"}); |
| testEval(schema2, "table1", "d,a,c", "select col1 between col2 and col3 from table1", new String[]{"f"}); |
| |
| // tests for asymmetric and symmetric |
| testEval(schema2, "table1", "b,a,c", "select col1 between col3 and col2 from table1", new String[]{"f"}); |
| testEval(schema2, "table1", "b,a,c", "select col1 between symmetric col3 and col2 from table1", new String[]{"t"}); |
| } |
| |
| @Test |
| public void testBetween2() throws TajoException { // for TAJO-249 |
| Schema schema3 = SchemaFactory.newV1(); |
| schema3.addColumn("date_a", INT4); |
| schema3.addColumn("date_b", INT4); |
| schema3.addColumn("date_c", INT4); |
| schema3.addColumn("date_d", INT4); |
| |
| String query = "select " + |
| "case " + |
| "when date_a BETWEEN 20130705 AND 20130715 AND ((date_b BETWEEN 20100101 AND 20120601) OR date_b > 20130715) " + |
| "AND (date_c < 20120601 OR date_c > 20130715) AND date_d > 20130715" + |
| "then 1 else 0 end from table1"; |
| |
| testEval(schema3, "table1", "20130715,20100102,20120525,20130716", query, new String [] {"1"}); |
| testEval(schema3, "table1", "20130716,20100102,20120525,20130716", query, new String [] {"0"}); |
| |
| // date_b |
| testEval(schema3, "table1", "20130715,20100102,20120525,20130716", query, new String [] {"1"}); |
| testEval(schema3, "table1", "20130715,20120602,20120525,20130716", query, new String [] {"0"}); |
| testEval(schema3, "table1", "20130715,20091201,20120525,20130716", query, new String [] {"0"}); |
| testEval(schema3, "table1", "20130715,20130716,20120525,20130716", query, new String [] {"1"}); |
| |
| // date_c |
| testEval(schema3, "table1", "20130715,20100102,20120525,20130716", query, new String [] {"1"}); |
| testEval(schema3, "table1", "20130715,20100102,20120602,20130716", query, new String [] {"0"}); |
| |
| testEval(schema3, "table1", "20130715,20100102,20130716,20130716", query, new String [] {"1"}); |
| testEval(schema3, "table1", "20130715,20100102,20130714,20130716", query, new String [] {"0"}); |
| |
| // date_d |
| testEval(schema3, "table1", "20130715,20100102,20120525,20130716", query, new String [] {"1"}); |
| testEval(schema3, "table1", "20130715,20100102,20120525,20130705", query, new String [] {"0"}); |
| } |
| |
| ////////////////////////////////////////////////////////////////// |
| // In Predicate |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testInPredicateWithConstant() throws TajoException { |
| Schema schema2 = SchemaFactory.newV1(); |
| schema2.addColumn("col1", TEXT); |
| schema2.addColumn("col2", TEXT); |
| schema2.addColumn("col3", TEXT); |
| |
| testEval(schema2, "table1", "a,b,c", "select col1 in ('a'), col2 in ('a', 'c') from table1", new String[]{"t","f"}); |
| testEval(schema2, "table1", "a,,c", "select col1 in ('a','b','c'), (col2 in ('a', 'c')) is null from table1", |
| new String[]{"t","t"}); |
| |
| testEval(schema2, |
| "table1", |
| "2014-03-21,2015-04-01,2016-04-01", |
| "select substr(col1,1,4) in ('2014','2015','2016'), substr(col1,6,2)::int4 in (1,2,3) from table1", |
| new String[]{"t", "t"}); |
| |
| // null handling test |
| testEval(schema2, |
| "table1", |
| "2014-03-21,,2015-04-01", |
| "select (substr(col2,1,4)::int4 in (2014,2015,2016)) is null from table1", |
| new String[]{"t"}); |
| } |
| |
| @Test |
| public void testInPredicateWithSimpleExprs() throws TajoException { |
| Schema schema2 = SchemaFactory.newV1(); |
| schema2.addColumn("col1", TEXT); |
| schema2.addColumn("col2", INT4); |
| schema2.addColumn("col3", TEXT); |
| |
| testEval(schema2, "table1", "abc,2,3", "select col1 in ('a'||'b'||'c'), col2 in (1 + 1, 2 * 10, 2003) from table1", |
| new String[]{"t","t"}); |
| |
| testEval(schema2, "table1", "abc,2,3", "select col1 in ('a'||'b'), col2 in ('1'::int, '2'::int, 3) from table1", |
| new String[]{"f","t"}); |
| |
| testEval(schema2, |
| "table1", |
| "abc,,3", |
| "select col1 in (reverse('cba')), (col2 in ('1'::int, '2'::int, 3)) is null from table1", |
| new String[]{"t","t"}); |
| } |
| |
| ////////////////////////////////////////////////////////////////// |
| // Null Predicate |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testIsNullPredicate() throws TajoException { |
| Schema schema1 = SchemaFactory.newV1(); |
| schema1.addColumn("col1", INT4); |
| schema1.addColumn("col2", INT4); |
| testEval(schema1, "table1", "123,", "select col1 is null, col2 is null as a from table1", |
| new String[]{"f", "t"}); |
| testEval(schema1, "table1", "123,", "select col1 is not null, col2 is not null as a from table1", |
| new String[]{"t", "f"}); |
| } |
| |
| @Test |
| public void testIsNullPredicateWithFunction() throws TajoException { |
| Schema schema2 = SchemaFactory.newV1(); |
| schema2.addColumn("col1", TEXT); |
| schema2.addColumn("col2", TEXT); |
| testEval(schema2, "table1", "_123,", "select ltrim(col1, '_') is null, upper(col2) is null as a from table1", |
| new String[]{"f", "t"}); |
| |
| testEval(schema2, "table1", "_123,", |
| "select ltrim(col1, '_') is not null, upper(col2) is not null as a from table1", new String[]{"t", "f"}); |
| } |
| |
| ////////////////////////////////////////////////////////////////// |
| // Boolean Test |
| ////////////////////////////////////////////////////////////////// |
| |
| @Test |
| public void testBooleanTest() throws TajoException { |
| testSimpleEval("select 1 < 3 is true", new String [] {"t"}); |
| testSimpleEval("select 1 < 3 is not true", new String [] {"f"}); |
| testSimpleEval("select 1 < 3 is false", new String [] {"f"}); |
| testSimpleEval("select 1 < 3 is not false", new String [] {"t"}); |
| |
| testSimpleEval("select not (1 < 3 is true)", new String [] {"f"}); |
| testSimpleEval("select not (1 < 3 is not true)", new String [] {"t"}); |
| testSimpleEval("select not (1 < 3 is false)", new String [] {"t"}); |
| testSimpleEval("select not (1 < 3 is not false)", new String [] {"f"}); |
| |
| testSimpleEval("select 1 > 3 is true", new String [] {"f"}); |
| testSimpleEval("select 1 > 3 is not true", new String [] {"t"}); |
| testSimpleEval("select 1 > 3 is false", new String [] {"t"}); |
| testSimpleEval("select 1 > 3 is not false", new String [] {"f"}); |
| |
| testSimpleEval("select not (1 > 3 is true)", new String [] {"t"}); |
| testSimpleEval("select not (1 > 3 is not true)", new String [] {"f"}); |
| testSimpleEval("select not (1 > 3 is false)", new String [] {"f"}); |
| testSimpleEval("select not (1 > 3 is not false)", new String [] {"t"}); |
| } |
| |
| @Test |
| public void testBooleanTestOnTable() throws TajoException { |
| Schema schema = SchemaFactory.newV1(); |
| schema.addColumn("col1", BOOLEAN); |
| schema.addColumn("col2", BOOLEAN); |
| testEval(schema, "table1", "t,f", "select col1 is true, col2 is false from table1", new String [] {"t", "t"}); |
| testEval(schema, "table1", "t,f", "select col1 is not true, col2 is not false from table1", |
| new String [] {"f", "f"}); |
| testEval(schema, "table1", "t,f", "select not col1 is not true, not col2 is not false from table1", |
| new String [] {"t", "t"}); |
| } |
| } |