def test_batch_shapes_a(self) -> None:
dc = DisplayConfig.from_default(type_color=False)
f1 = Frame.from_dict({'a':[1,2,3], 'b':[2,4,6], 'group': ['x','z','z']})
b1 = Batch(f1.iter_group_items('group'))[['a', 'b']]
self.assertEqual(b1.shapes.to_pairs(),
(('x', (1, 2)), ('z', (2, 2)))
)
def test_batch_display_a(self) -> None:
dc = DisplayConfig.from_default(type_color=False)
f1 = Frame.from_dict({'a':[1,2,3], 'b':[2,4,6], 'group': ['x','z','z']})
gi = f1.iter_group_items('group')
d1 = Batch(gi)[['a', 'b']].display(dc)
self.assertEqual(d1.to_rows(),
['<Batch>', '<Index>',
'x <Frame>',
'z <Frame>',
'<<U1> <object>'
])