def test_having_simple(self):
tableset = TableSet(self.tables.values(), self.tables.keys(), key_name='test')
new_tableset = tableset.having([
('count', Count())
], lambda t: t['count'] < 3)
self.assertIsInstance(new_tableset, TableSet)
self.assertSequenceEqual(new_tableset.keys(), ['table3'])
self.assertIs(new_tableset.values()[0], tableset['table3'])
self.assertEqual(new_tableset.key_name, 'test')
def test_having_complex(self):
tableset = TableSet(self.tables.values(), self.tables.keys(), key_name='test')
new_tableset = tableset.having([
('count', Count()),
('number_sum', Sum('number'))
], lambda t: t['count'] >= 3 and t['number_sum'] > 6)
self.assertIsInstance(new_tableset, TableSet)
self.assertSequenceEqual(new_tableset.keys(), ['table2'])
self.assertIs(new_tableset.values()[0], tableset['table2'])
self.assertEqual(new_tableset.key_name, 'test')
def test_having_simple(self):
tableset = TableSet(self.tables.values(),
self.tables.keys(),
key_name='test')
new_tableset = tableset.having([('count', Count())],
lambda t: t['count'] < 3)
self.assertIsInstance(new_tableset, TableSet)
self.assertSequenceEqual(new_tableset.keys(), ['table3'])
self.assertIs(new_tableset.values()[0], tableset['table3'])
self.assertEqual(new_tableset.key_name, 'test')
def test_having_complex(self):
tableset = TableSet(self.tables.values(),
self.tables.keys(),
key_name='test')
new_tableset = tableset.having(
[('count', Count()), ('number_sum', Sum('number'))],
lambda t: t['count'] >= 3 and t['number_sum'] > 6)
self.assertIsInstance(new_tableset, TableSet)
self.assertSequenceEqual(new_tableset.keys(), ['table2'])
self.assertIs(new_tableset.values()[0], tableset['table2'])
self.assertEqual(new_tableset.key_name, 'test')
