Skip to content

Commit ed3a044

Browse files
committed
BUG: PeriodIndex fails to handle NA, rather than putting NaT in its place (pandas-dev#46673)
1 parent 841bb6e commit ed3a044

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pandas/tests/frame/methods/test_dtypes.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -83,15 +83,15 @@ def test_dtypes_are_correct_after_column_slice(self):
8383
@pytest.mark.parametrize(
8484
"data",
8585
[
86-
DataFrame({"id": [1, 2, 3, 4], "test": [True, pd.NA, pd.NA, False]}),
87-
DataFrame({"id": [1, 2, 3, 4], "test": [True, pd.NA, True, False]}),
86+
[True, pd.NA, pd.NA, False],
87+
[True, pd.NA, True, False],
8888
],
8989
)
9090
def test_dtypes_are_correct_after_groupby_last(self, data):
9191
# GH46409
92-
data = data.convert_dtypes()
93-
result = data.groupby("id").last().test
94-
expected = data.set_index("id").test
92+
df = DataFrame({"id": [1, 2, 3, 4], "test": data}).convert_dtypes()
93+
result = df.groupby("id").last().test
94+
expected = df.set_index("id").test
9595
assert result.dtype == pd.BooleanDtype()
9696
tm.assert_series_equal(expected, result)
9797

0 commit comments

Comments
 (0)