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
def test_pandas(test_data): """Test creating chart data source from existing dataframe.""" ds = ChartDataSource.from_data(test_data.pd_data) assert len(ds.columns) == 2 assert len(ds.index) == 4 ds = ChartDataSource(test_data.pd_data) assert len(ds.columns) == 2 assert len(ds.index) == 4