Skip to content

Commit b257ccf

Browse files
thomasqueirozbandrekwrHenryRocha
committed
TST: add to_datetime of NaT values with utc=True
Co-authored-by: André Elimelek de Weber (andrekwr) <[email protected]> Co-authored-by: Henry Rocha (HenryRocha) <[email protected]>
1 parent 381dd41 commit b257ccf

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

pandas/tests/tools/test_to_datetime.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
timedelta,
88
)
99
from decimal import Decimal
10+
from functools import partial
1011
import locale
1112

1213
from dateutil.parser import parse
@@ -1014,6 +1015,36 @@ def test_convert_object_to_datetime_with_cache(
10141015
)
10151016
tm.assert_series_equal(result_series, expected_series)
10161017

1018+
@pytest.mark.parametrize("cache", [True, False])
1019+
@pytest.mark.parametrize(
1020+
"constructor",
1021+
(
1022+
partial(Series, dtype="object"),
1023+
list,
1024+
np.array,
1025+
),
1026+
)
1027+
@pytest.mark.parametrize(
1028+
"datetimelikes,expected_values",
1029+
(
1030+
(
1031+
(datetime(3000, 1, 1, 0, 0, 0, 0, pytz.UTC),) * (start_caching_at + 1),
1032+
(NaT,) * (start_caching_at + 1),
1033+
),
1034+
),
1035+
)
1036+
def test_convert_object_to_datetime_nat_utc(
1037+
self, cache, datetimelikes, expected_values, constructor
1038+
):
1039+
# GH#43732
1040+
res = constructor(datetimelikes)
1041+
result_series = Series(to_datetime(res, errors="coerce", cache=cache, utc=True))
1042+
expected_series = Series(
1043+
expected_values,
1044+
dtype="datetime64[ns, UTC]",
1045+
)
1046+
tm.assert_series_equal(result_series, expected_series)
1047+
10171048
@pytest.mark.parametrize(
10181049
"date, format",
10191050
[

0 commit comments

Comments
 (0)