From c1718401ecaf45eeaed2e79095f2a089fa86f547 Mon Sep 17 00:00:00 2001 From: Loic Diridollou Date: Sun, 12 Jan 2025 16:01:32 -0500 Subject: [PATCH] GH1095 Add list of string to to_timedelta --- pandas-stubs/core/tools/timedeltas.pyi | 2 +- tests/test_timefuncs.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/pandas-stubs/core/tools/timedeltas.pyi b/pandas-stubs/core/tools/timedeltas.pyi index 38ed0c074..f5a733c27 100644 --- a/pandas-stubs/core/tools/timedeltas.pyi +++ b/pandas-stubs/core/tools/timedeltas.pyi @@ -33,7 +33,7 @@ def to_timedelta( def to_timedelta( arg: ( SequenceNotStr - | Sequence[float | timedelta] + | Sequence[float | timedelta | str] | tuple[str | float | timedelta, ...] | range | ArrayLike diff --git a/tests/test_timefuncs.py b/tests/test_timefuncs.py index 8f9424713..fba3ade3e 100644 --- a/tests/test_timefuncs.py +++ b/tests/test_timefuncs.py @@ -1425,3 +1425,10 @@ def test_DatetimeIndex_sub_timedelta() -> None: ), pd.DatetimeIndex, ) + + +def test_datetimeindex_as_unit() -> None: + """Test that we can pass a list of str for `to_timedelta` GH1095.""" + tdelta_idx = pd.to_timedelta(["1 day 3 min 2 us 42 ns"]) + check(assert_type(tdelta_idx, pd.TimedeltaIndex), pd.TimedeltaIndex) + check(assert_type(tdelta_idx.as_unit("s"), pd.TimedeltaIndex), pd.TimedeltaIndex)