22 lines
574 B
Python
22 lines
574 B
Python
|
import pytest
|
||
|
|
|
||
|
|
import pandas as pd
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"to_concat_dtypes, result_dtype",
|
||
|
|
[
|
||
|
|
(["Float64", "Float64"], "Float64"),
|
||
|
|
(["Float32", "Float64"], "Float64"),
|
||
|
|
(["Float32", "Float32"], "Float32"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_concat_series(to_concat_dtypes, result_dtype):
|
||
|
|
|
||
|
|
result = pd.concat([pd.Series([1, 2, pd.NA], dtype=t) for t in to_concat_dtypes])
|
||
|
|
expected = pd.concat([pd.Series([1, 2, pd.NA], dtype=object)] * 2).astype(
|
||
|
|
result_dtype
|
||
|
|
)
|
||
|
|
tm.assert_series_equal(result, expected)
|