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)