@@ -3004,31 +3004,37 @@ def test_groupby_agg_namedagg_with_duplicate_columns():
3004
3004
3005
3005
tm .assert_frame_equal (result , expected )
3006
3006
3007
+
3007
3008
class MyDataFrame (DataFrame ):
3008
3009
@property
3009
3010
def _constructor (self ):
3010
3011
return MyDataFrame
3011
3012
3012
- @pytest .mark .parametrize ("data, agg_dict, expected" , [
3013
- pytest .param (
3014
- {"A" : [1 , 1 , 2 , 2 ], "B" : [1 , 2 , 3 , 4 ]},
3015
- {"B" : "sum" },
3016
- DataFrame ({"B" : [3 , 7 ]}, index = Index ([1 , 2 ], name = "A" ))
3017
- ),
3018
- pytest .param (
3019
- {"A" : [1 , 1 , 2 , 2 ], "B" : [1 , 2 , 3 , 4 ], "C" : [4 , 3 , 2 , 1 ]},
3020
- {"B" : "sum" , "C" : "mean" },
3021
- DataFrame ({"B" : [3 , 7 ], "C" : [3.5 , 1.5 ]}, index = Index ([1 , 2 ], name = "A" ))
3022
- ),
3023
- ])
3013
+
3014
+ @pytest .mark .parametrize (
3015
+ "data, agg_dict, expected" ,
3016
+ [
3017
+ pytest .param (
3018
+ {"A" : [1 , 1 , 2 , 2 ], "B" : [1 , 2 , 3 , 4 ]},
3019
+ {"B" : "sum" },
3020
+ DataFrame ({"B" : [3 , 7 ]}, index = Index ([1 , 2 ], name = "A" )),
3021
+ ),
3022
+ pytest .param (
3023
+ {"A" : [1 , 1 , 2 , 2 ], "B" : [1 , 2 , 3 , 4 ], "C" : [4 , 3 , 2 , 1 ]},
3024
+ {"B" : "sum" , "C" : "mean" },
3025
+ DataFrame ({"B" : [3 , 7 ], "C" : [3.5 , 1.5 ]}, index = Index ([1 , 2 ], name = "A" )),
3026
+ ),
3027
+ ],
3028
+ )
3024
3029
def test_groupby_agg_preserves_subclass (data , agg_dict , expected ):
3025
3030
# GH#59667
3026
3031
df = MyDataFrame (data )
3027
3032
result = df .groupby ("A" ).agg (agg_dict )
3028
-
3033
+
3029
3034
assert isinstance (result , MyDataFrame )
3030
3035
tm .assert_frame_equal (result , expected )
3031
3036
3037
+
3032
3038
def test_groupby_multi_index_codes ():
3033
3039
# GH#54347
3034
3040
df = DataFrame (
0 commit comments