def test_compile_targets_named(self): # Test the wildcard expansion. query = self.compile("SELECT length(account), account as a, date;") self.assertEqual( [qc.EvalTarget(qe.Length([qe.AccountColumn()]), 'length_account', False), qc.EvalTarget(qe.AccountColumn(), 'a', False), qc.EvalTarget(qe.DateColumn(), 'date', False)], query.c_targets)
def test_get_columns_and_aggregates(self): # Simple column. c_query = qe.PositionColumn() columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((1, 0), (len(columns), len(aggregates))) self.assertFalse(qc.is_aggregate(c_query)) # Multiple columns. c_query = qc.EvalAnd(qe.PositionColumn(), qe.DateColumn()) columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((2, 0), (len(columns), len(aggregates))) self.assertFalse(qc.is_aggregate(c_query)) # Simple aggregate. c_query = qe.SumPosition([qe.PositionColumn()]) columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((0, 1), (len(columns), len(aggregates))) self.assertTrue(qc.is_aggregate(c_query)) # Multiple aggregates. c_query = qc.EvalAnd(qe.First([qe.AccountColumn()]), qe.Last([qe.AccountColumn()])) columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((0, 2), (len(columns), len(aggregates))) self.assertTrue(qc.is_aggregate(c_query)) # Simple non-aggregate function. c_query = qe.Length([qe.AccountColumn()]) columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((1, 0), (len(columns), len(aggregates))) self.assertFalse(qc.is_aggregate(c_query)) # Mix of column and aggregates (this is used to detect this illegal case). c_query = qc.EvalAnd(qe.Length([qe.AccountColumn()]), qe.SumPosition([qe.PositionColumn()])) columns, aggregates = qc.get_columns_and_aggregates(c_query) self.assertEqual((1, 1), (len(columns), len(aggregates))) self.assertTrue(qc.is_aggregate(c_query))
def test_compile_EvalLength(self): with self.assertRaises(qc.CompilationError): qe.Length([qc.EvalConstant(17)]) c_length = qe.Length([qc.EvalConstant('testing')]) self.assertEqual(int, c_length.dtype)