Skip to content

Commit bf69b37

Browse files
ftrihardjoluckyvs1
authored andcommitted
TST: Inconsistent behavior of .replace() in Int64 series with NA (pandas-dev#38693)
1 parent 4463bd0 commit bf69b37

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

pandas/tests/series/methods/test_replace.py

+9
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,15 @@ def test_replace_with_dict_with_bool_keys(self):
208208
expected = pd.Series(["yes", False, "yes"])
209209
tm.assert_series_equal(result, expected)
210210

211+
def test_replace_Int_with_na(self, any_nullable_int_dtype):
212+
# GH 38267
213+
result = pd.Series([0, None], dtype=any_nullable_int_dtype).replace(0, pd.NA)
214+
expected = pd.Series([pd.NA, pd.NA], dtype=any_nullable_int_dtype)
215+
tm.assert_series_equal(result, expected)
216+
result = pd.Series([0, 1], dtype=any_nullable_int_dtype).replace(0, pd.NA)
217+
result.replace(1, pd.NA, inplace=True)
218+
tm.assert_series_equal(result, expected)
219+
211220
def test_replace2(self):
212221
N = 100
213222
ser = pd.Series(np.fabs(np.random.randn(N)), tm.makeDateIndex(N), dtype=object)

0 commit comments

Comments
 (0)