1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
| import pytest
|
| import pandas.util._test_decorators as td
|
| from pandas import (
| Interval,
| Period,
| Series,
| Timedelta,
| Timestamp,
| )
|
|
| @pytest.mark.parametrize(
| "values, dtype, expected_dtype",
| (
| ([1], "int64", int),
| ([1], "Int64", int),
| ([1.0], "float64", float),
| ([1.0], "Float64", float),
| (["abc"], "object", str),
| (["abc"], "string", str),
| ([Interval(1, 3)], "interval", Interval),
| ([Period("2000-01-01", "D")], "period[D]", Period),
| ([Timedelta(days=1)], "timedelta64[ns]", Timedelta),
| ([Timestamp("2000-01-01")], "datetime64[ns]", Timestamp),
| pytest.param([1], "int64[pyarrow]", int, marks=td.skip_if_no("pyarrow")),
| pytest.param([1.0], "float64[pyarrow]", float, marks=td.skip_if_no("pyarrow")),
| pytest.param(["abc"], "string[pyarrow]", str, marks=td.skip_if_no("pyarrow")),
| ),
| )
| def test_tolist_scalar_dtype(values, dtype, expected_dtype):
| # GH49890
| ser = Series(values, dtype=dtype)
| result_dtype = type(ser.tolist()[0])
| assert result_dtype == expected_dtype
|
|
