@@ -1125,27 +1125,6 @@ def test_comp_masked_numpy(self, masked_dtype, comparison_op):
1125
1125
expected = pd .Series (exp , dtype = ArrowDtype (pa .bool_ ()))
1126
1126
tm .assert_series_equal (result , expected )
1127
1127
1128
- def test_groupby_agg_extension (self , data_for_grouping ):
1129
- # GH#38980 groupby agg on extension type fails for non-numeric types
1130
- df = pd .DataFrame ({"A" : [1 , 1 , 2 , 2 , 3 , 3 , 1 , 4 ], "B" : data_for_grouping })
1131
-
1132
- expected_df = pd .DataFrame (
1133
- {"A" : [1 , 1 , 2 , 2 , 3 , 3 , 1 , 4 ], "B" : data_for_grouping }
1134
- )
1135
- expected = expected_df .iloc [[0 , 2 , 4 , 7 ]]
1136
- expected = expected .set_index ("A" )
1137
- expected_dtype = expected ["B" ].dtype
1138
- expected ["B" ] = expected ["B" ].astype (expected_dtype )
1139
-
1140
- result = df .groupby ("A" ).agg ({"B" : "first" })
1141
- tm .assert_frame_equal (result , expected )
1142
-
1143
- result = df .groupby ("A" ).agg ("first" )
1144
- tm .assert_frame_equal (result , expected )
1145
-
1146
- result = df .groupby ("A" ).first ()
1147
- tm .assert_frame_equal (result , expected )
1148
-
1149
1128
1150
1129
class TestLogicalOps :
1151
1130
"""Various Series and DataFrame logical ops methods."""
0 commit comments