Skip to content

Commit 5512e29

Browse files
committed
test for nested EA
1 parent 9d20b61 commit 5512e29

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/series/test_explode.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,3 +91,18 @@ def test_typical_usecase():
9191
{"var2": [1, 1, 1, 2, 2, 2], "var1": list("abcdef")}, index=[0, 0, 0, 1, 1, 1]
9292
)
9393
tm.assert_frame_equal(result, expected, check_like=True)
94+
95+
96+
def test_nested_EA():
97+
# a nested EA array
98+
s = pd.Series(
99+
[
100+
pd.date_range("20170101", periods=3, tz="UTC"),
101+
pd.date_range("20170104", periods=3, tz="UTC"),
102+
]
103+
)
104+
result = s.explode()
105+
expected = pd.Series(
106+
pd.date_range("20170101", periods=6, tz="UTC"), index=[0, 0, 0, 1, 1, 1]
107+
)
108+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)