Example #1
0
 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)
Example #2
0
 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)