def test_grouped_as_dicts(self): b = Box(get_lst4()) keys = ['a', 'c'] res = b.grouped(keys, as_dicts=True) expected = [{ 'labels': { 'b': 1 }, 'values': { 'a': [1, 2], 'c': [4, 6] } }, { 'labels': { 'b': 2 }, 'values': { 'a': [1, 2], 'c': [5, 7] } }] self.assertListEqual(res, expected)
def test_grouped_as_lists(self): b = Box(get_lst4()) keys = ['a', 'c'] res = b.grouped(keys, as_dicts=False) expected = [[[1, 2], [4, 6], {'b': 1}], [[1, 2], [5, 7], {'b': 2}]] self.assertListEqual(res, expected)