Skip to content

Commit a1d73f5

Browse files
author
Kei
committed
Update tests
1 parent e1ccef6 commit a1d73f5

File tree

1 file changed

+1
-12
lines changed

1 file changed

+1
-12
lines changed

pandas/tests/extension/test_arrow.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1129,23 +1129,12 @@ def test_groupby_agg_extension(self, data_for_grouping):
11291129
# GH#38980 groupby agg on extension type fails for non-numeric types
11301130
df = pd.DataFrame({"A": [1, 1, 2, 2, 3, 3, 1, 4], "B": data_for_grouping})
11311131

1132-
def expected_inferred_result_dtype(dtype):
1133-
pa_dtype = dtype.pyarrow_dtype
1134-
if pa.types.is_date(pa_dtype):
1135-
return "date32[day][pyarrow]"
1136-
elif pa.types.is_time(pa_dtype):
1137-
return "time64[us][pyarrow]"
1138-
elif pa.types.is_decimal(pa_dtype):
1139-
return ArrowDtype(pa.decimal128(4, 3))
1140-
else:
1141-
return dtype
1142-
11431132
expected_df = pd.DataFrame(
11441133
{"A": [1, 1, 2, 2, 3, 3, 1, 4], "B": data_for_grouping}
11451134
)
11461135
expected = expected_df.iloc[[0, 2, 4, 7]]
11471136
expected = expected.set_index("A")
1148-
expected_dtype = expected_inferred_result_dtype(expected["B"].dtype)
1137+
expected_dtype = expected["B"].dtype
11491138
expected["B"] = expected["B"].astype(expected_dtype)
11501139

11511140
result = df.groupby("A").agg({"B": "first"})

0 commit comments

Comments
 (0)