Skip to content

ENH: Improve general datetime functions #352

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 15 commits into from
Oct 4, 2022
Merged

ENH: Improve general datetime functions #352

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
e7d5294
TYP: Improve infer_freq
bashtage Oct 3, 2022
be6c87d
TYP: Imrprove timedelta_range
bashtage Oct 3, 2022
27fef79
TYP: Imrprove period_range
bashtage Oct 3, 2022
68e41e1
TYP: Improve date_range and bdate_range
bashtage Oct 3, 2022
d360eb6
TYP: Improve to_timedelta
bashtage Oct 3, 2022
539c842
TYP: Improve to_datetime
bashtage Oct 3, 2022
234fa44
Small fixes
bashtage Oct 3, 2022
47b9a6d
TST: Add tests for to_timedelta
bashtage Oct 3, 2022
028c753
Merge remote-tracking branch 'upstream/main' into gen-func-datetime
bashtage Oct 3, 2022
94b94e5
TST/ENH: Add tests and overload for bdate_range
bashtage Oct 3, 2022
860cad7
TST/ENH: Add tests and improve period_range
bashtage Oct 3, 2022
c447e21
TST: Add tests
Oct 4, 2022
b167110
TST: Add check and assert type, add more tests
Oct 4, 2022
fda9787
CLN: Remove unnecessary TODO
Oct 4, 2022
9786946
ENH: Improve timestamp convertible
bashtage Oct 4, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
TST: Add check and assert type, add more tests
  • Loading branch information
Kevin Sheppard committed Oct 4, 2022
commit b167110af96cac0fc222b18236ce393c4b7de18d
135 changes: 97 additions & 38 deletions tests/test_timefuncs.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -542,44 +542,44 @@ def test_types_to_numpy() -> None:


def test_to_timdelta_units() -> None:
pd.to_timedelta(1, "W")
pd.to_timedelta(1, "w")
pd.to_timedelta(1, "D")
pd.to_timedelta(1, "d")
pd.to_timedelta(1, "days")
pd.to_timedelta(1, "day")
pd.to_timedelta(1, "hours")
pd.to_timedelta(1, "hour")
pd.to_timedelta(1, "hr")
pd.to_timedelta(1, "h")
pd.to_timedelta(1, "m")
pd.to_timedelta(1, "minute")
pd.to_timedelta(1, "min")
pd.to_timedelta(1, "minutes")
pd.to_timedelta(1, "t")
pd.to_timedelta(1, "s")
pd.to_timedelta(1, "seconds")
pd.to_timedelta(1, "sec")
pd.to_timedelta(1, "second")
pd.to_timedelta(1, "ms")
pd.to_timedelta(1, "milliseconds")
pd.to_timedelta(1, "millisecond")
pd.to_timedelta(1, "milli")
pd.to_timedelta(1, "millis")
pd.to_timedelta(1, "l")
pd.to_timedelta(1, "us")
pd.to_timedelta(1, "microseconds")
pd.to_timedelta(1, "microsecond")
pd.to_timedelta(1, "µs")
pd.to_timedelta(1, "micro")
pd.to_timedelta(1, "micros")
pd.to_timedelta(1, "u")
pd.to_timedelta(1, "ns")
pd.to_timedelta(1, "nanoseconds")
pd.to_timedelta(1, "nano")
pd.to_timedelta(1, "nanos")
pd.to_timedelta(1, "nanosecond")
pd.to_timedelta(1, "n")
check(assert_type(pd.to_timedelta(1, "W"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "w"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "D"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "d"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "days"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "day"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "hours"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "hour"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "hr"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "h"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "m"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "minute"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "min"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "minutes"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "t"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "s"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "seconds"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "sec"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "second"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "ms"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "milliseconds"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "millisecond"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "milli"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "millis"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "l"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "us"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "microseconds"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "microsecond"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "µs"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "micro"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "micros"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "u"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "ns"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "nanoseconds"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "nano"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "nanos"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "nanosecond"), pd.Timedelta), pd.Timedelta)
check(assert_type(pd.to_timedelta(1, "n"), pd.Timedelta), pd.Timedelta)


def test_to_timedelta_scalar() -> None:
Expand Down Expand Up @@ -844,3 +844,62 @@ def test_to_datetime_array() -> None:
),
pd.DatetimeIndex,
)


def test_timedelta_range() -> None:
check(
assert_type(
pd.timedelta_range(
pd.Timedelta(1, unit="D"), pd.Timedelta(10, unit="D"), periods=10
),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range(dt.timedelta(1), dt.timedelta(10), periods=10),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range(
np.timedelta64(86400000000000),
np.timedelta64(864000000000000),
periods=10,
),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range("1 day", "10 days", periods=10), pd.TimedeltaIndex
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range(
np.int64(86400000000000), np.int64(864000000000000), periods=10
),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range(86400000000000, 864000000000000, periods=10),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)
check(
assert_type(
pd.timedelta_range(86400000000000.0, 864000000000000.0, periods=10),
pd.TimedeltaIndex,
),
pd.TimedeltaIndex,
)