1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
| 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)
|
|