Skip to content

Commit c6d2f92

Browse files
authored
TST: Verify mask's return keeps dtype (#50460)
1 parent e00b75b commit c6d2f92

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/frame/indexing/test_mask.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from pandas import (
88
NA,
99
DataFrame,
10+
Float64Dtype,
1011
Series,
1112
StringDtype,
1213
Timedelta,
@@ -130,3 +131,13 @@ def test_mask_where_dtype_timedelta():
130131
[np.nan, np.nan, np.nan, Timedelta("3 day"), Timedelta("4 day")]
131132
)
132133
tm.assert_frame_equal(df.where(df > Timedelta(2, unit="d")), expected)
134+
135+
136+
def test_mask_return_dtype():
137+
# GH#50488
138+
ser = Series([0.0, 1.0, 2.0, 3.0], dtype=Float64Dtype())
139+
cond = ~ser.isna()
140+
other = Series([True, False, True, False])
141+
excepted = Series([1.0, 0.0, 1.0, 0.0], dtype=ser.dtype)
142+
result = ser.mask(cond, other)
143+
tm.assert_series_equal(result, excepted)

0 commit comments

Comments
 (0)