From c37b55d622963998cd163da1c8a262b04d5cbaa5 Mon Sep 17 00:00:00 2001 From: Kaiqi Dong Date: Sun, 2 Feb 2020 00:10:40 +0100 Subject: [PATCH] Backport PR #31545: BUG&TST: df.replace fail after converting to new dtype --- pandas/tests/frame/methods/test_replace.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pandas/tests/frame/methods/test_replace.py b/pandas/tests/frame/methods/test_replace.py index aa91e7a489356..92b74c4409d7d 100644 --- a/pandas/tests/frame/methods/test_replace.py +++ b/pandas/tests/frame/methods/test_replace.py @@ -1356,3 +1356,10 @@ def test_replace_replacer_dtype(self, replacer): result = df.replace({"a": replacer, "b": replacer}) expected = pd.DataFrame([replacer]) tm.assert_frame_equal(result, expected) + + def test_replace_after_convert_dtypes(self): + # GH31517 + df = pd.DataFrame({"grp": [1, 2, 3, 4, 5]}, dtype="Int64") + result = df.replace(1, 10) + expected = pd.DataFrame({"grp": [10, 2, 3, 4, 5]}, dtype="Int64") + tm.assert_frame_equal(result, expected)