def test_prepare_columns(expressions, cs):
tree = SQLTree()
tree.expressions = expressions
actual_cs = prepare_columns(tree)
print('Expected: %s' % cs)
print('Actual: %s' % actual_cs)
assert actual_cs == cs
def test_group_result_set(rs, row, expressions, result):
tree = SQLTree()
tree.expressions = expressions
actual = group_result_set(etcdb_count, rs, row, tree, 0)
print('Expected: %s' % result)
print('Actual: %s' % actual)
assert actual == result
def test_expr_in_simple_expr():
tree = SQLTree()
tree.expressions = [(('bool_primary', ('predicate',
('bit_expr',
('simple_expr',
('expr', ('bool_primary',
('predicate',
('bit_expr',
('simple_expr',
('literal',
'1')))))))))), 'a')]
assert eval_expr(None, tree.expressions[0][0]) == ('1', '1')
def test_eval_row(cs, row, expressions, result):
tree = SQLTree()
tree.expressions = expressions
assert eval_row(cs, row, tree) == result