Skip to content

Commit 574caf5

Browse files
amgcctwoertwein
authored andcommitted
Add test
1 parent 9aefaa4 commit 574caf5

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/test_pandas.py

+15
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,21 @@ def test_types_to_datetime() -> None:
4949
)
5050

5151

52+
def test_types_concat_none() -> None:
53+
"""Test concatenation with None values."""
54+
s1: pd.Series | None = None
55+
s2: pd.Series = pd.Series([7, -5, 10])
56+
57+
check(assert_type(pd.concat([s1, s2]), pd.Series), pd.Series)
58+
59+
df1: pd.DataFrame | None = None
60+
df2: pd.DataFrame = pd.DataFrame({"a": [7, -5, 10]})
61+
62+
check(assert_type(pd.concat([df1, df2]), pd.DataFrame), pd.DataFrame)
63+
64+
check(assert_type(pd.concat([s1, df1, df2]), pd.DataFrame), pd.DataFrame)
65+
66+
5267
def test_types_concat() -> None:
5368
s: pd.Series = pd.Series([0, 1, -10])
5469
s2: pd.Series = pd.Series([7, -5, 10])

0 commit comments

Comments
 (0)