def test_groupby_to_scalar_to_series_1(self): s = pd.Series([1, 2, 3, 4]) labels = ['a', 'a', 'b', 'b'] benchmark = s.groupby(labels).apply(max) result = pandas_easy.groupby_to_scalar_to_series( s, max, 1, by=labels) assert_series_equal(result, benchmark)
def test_groupby_to_scalar_to_series_1(self): df = pd.DataFrame({'a': [6, 2, 2], 'b': [4, 5, 6]}) benchmark = df.groupby('a').apply(max) result = pandas_easy.groupby_to_scalar_to_series(df, max, 1, by='a') assert_series_equal(result, benchmark)