Skip to content

Commit b48c9da

Browse files
committed
Add test for 26468
1 parent 8b05fe3 commit b48c9da

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/series/indexing/test_setitem.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
Timestamp,
1212
date_range,
1313
period_range,
14+
array,
15+
DataFrame,
1416
)
1517
from pandas.core.indexing import IndexingError
1618
import pandas.testing as tm
@@ -126,6 +128,16 @@ def test_setitem_boolean_different_order(self, string_series):
126128

127129
tm.assert_series_equal(copy, expected)
128130

131+
def test_setitem_boolean_ea_type(self):
132+
# GH: 26468
133+
df = DataFrame(
134+
{"a": [0, 0, np.nan, np.nan], "b": array(range(4), dtype="Int64")}
135+
)
136+
s = Series(array([1] * 4, dtype="Int64"))
137+
s[df["a"].isna()] = df.loc[df["a"].isna(), "b"]
138+
expected = Series([1, 1, 2, 3], dtype="Int64")
139+
tm.assert_series_equal(s, expected)
140+
129141

130142
class TestSetitemViewCopySemantics:
131143
def test_setitem_invalidates_datetime_index_freq(self):

0 commit comments

Comments
 (0)