def test_aggregate(self): aggregates = { "asc sum": { "column": "asc_idx", "operator": "sum" }, "asc q2": { "column": "asc_idx", "operator": "percentile", "options": { "q": 75 }, }, "desc q1": { "column": "desc_idx", "operator": "percentile", "options": { "q": 25 }, }, } df = proc.aggregate(df=categories_df, groupby=["constant"], aggregates=aggregates) self.assertListEqual(df.columns.tolist(), ["constant", "asc sum", "asc q2", "desc q1"]) self.assertEqual(series_to_list(df["asc sum"])[0], 5050) self.assertEqual(series_to_list(df["asc q2"])[0], 75) self.assertEqual(series_to_list(df["desc q1"])[0], 25)
def test_aggregate(): aggregates = { "asc sum": { "column": "asc_idx", "operator": "sum" }, "asc q2": { "column": "asc_idx", "operator": "percentile", "options": { "q": 75 }, }, "desc q1": { "column": "desc_idx", "operator": "percentile", "options": { "q": 25 }, }, } df = aggregate(df=categories_df, groupby=["constant"], aggregates=aggregates) assert df.columns.tolist() == ["constant", "asc sum", "asc q2", "desc q1"] assert series_to_list(df["asc sum"])[0] == 5050 assert series_to_list(df["asc q2"])[0] == 75 assert series_to_list(df["desc q1"])[0] == 25