Skip to content

Commit 8c7fa53

Browse files
author
Khor Chean Wei
authored
Test series replace different numpy int (#45797)
1 parent 23f8401 commit 8c7fa53

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/series/methods/test_replace.py

+11
Original file line numberDiff line numberDiff line change
@@ -668,3 +668,14 @@ def test_replace_regex_dtype_series(self, regex):
668668
expected = pd.Series([1])
669669
result = series.replace(to_replace="0", value=1, regex=regex)
670670
tm.assert_series_equal(result, expected)
671+
672+
def test_replace_different_int_types(self, any_int_numpy_dtype):
673+
# GH#45311
674+
labs = pd.Series([1, 1, 1, 0, 0, 2, 2, 2], dtype=any_int_numpy_dtype)
675+
676+
maps = pd.Series([0, 2, 1], dtype=any_int_numpy_dtype)
677+
map_dict = {old: new for (old, new) in zip(maps.values, maps.index)}
678+
679+
result = labs.replace(map_dict)
680+
expected = labs.replace({0: 0, 2: 1, 1: 2})
681+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)