Skip to content

Commit 760c2fa

Browse files
authored
TST: DataFrame.at on categorical with missing (#38085)
1 parent c26ac5d commit 760c2fa

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

pandas/tests/indexing/test_at.py

+18-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import numpy as np
44
import pytest
55

6-
from pandas import DataFrame, Series, Timestamp
6+
from pandas import CategoricalDtype, DataFrame, Series, Timestamp
77
import pandas._testing as tm
88

99

@@ -26,6 +26,23 @@ def test_at_setitem_mixed_index_assignment(self):
2626
ser.at[1] = 22
2727
assert ser.iat[3] == 22
2828

29+
def test_at_setitem_categorical_missing(self):
30+
df = DataFrame(
31+
index=range(3), columns=range(3), dtype=CategoricalDtype(["foo", "bar"])
32+
)
33+
df.at[1, 1] = "foo"
34+
35+
expected = DataFrame(
36+
[
37+
[np.nan, np.nan, np.nan],
38+
[np.nan, "foo", np.nan],
39+
[np.nan, np.nan, np.nan],
40+
],
41+
dtype=CategoricalDtype(["foo", "bar"]),
42+
)
43+
44+
tm.assert_frame_equal(df, expected)
45+
2946

3047
class TestAtSetItemWithExpansion:
3148
def test_at_setitem_expansion_series_dt64tz_value(self, tz_naive_fixture):

0 commit comments

Comments
 (0)