def test_filter_grouping_average(self):
# Only one of the three groupings meets condition
expected = [[0, 1]]
cond = 'PROMEDIO(col2) < 3'
grouping = [[0, 1], [2, 3], [4, 5]]
test = consult.filter_grps(cond, self.dtb1, grouping)
# Assert that there are two groupings
self.assertEqual(len(test), 1)
self.assertEqual(test, expected)
def test_filter_grouping_min(self):
# Only two of the three groupings meets condition
expected = [[2, 3], [4, 5]]
cond = 'MIN(col2) >= 2'
grouping = [[0, 1], [2, 3], [4, 5]]
test = consult.filter_grps(cond, self.dtb1, grouping)
# Assert that there are two groupings
self.assertEqual(len(test), 2)
self.assertEqual(test, expected)
