def test_groupby(test_data):
ds = ChartDataSource(df=test_data.auto_data)
groups = list(ds.groupby(**test_data.single_col_spec))
assert len(groups) == 5
ds = ChartDataSource(df=test_data.auto_data)
groups = list(ds.groupby(**test_data.multi_col_spec))
assert len(groups) == 9
