Skip to content

Commit 29b26a4

Browse files
nuztalgiaDaniel Gillet
authored andcommitted
Clean up some type annotations in time tests.
1 parent 1e97834 commit 29b26a4

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

tests/test_time.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,9 @@ def test_naturaldelta(test_input: float | dt.timedelta, expected: str) -> None:
170170
("NaN", "NaN"),
171171
],
172172
)
173-
def test_naturaltime(test_input: dt.datetime, expected: str) -> None:
173+
def test_naturaltime(
174+
test_input: dt.datetime | dt.timedelta | float, expected: str
175+
) -> None:
174176
assert humanize.naturaltime(test_input) == expected
175177

176178

@@ -211,7 +213,9 @@ def test_naturaltime(test_input: dt.datetime, expected: str) -> None:
211213
("NaN", "NaN"),
212214
],
213215
)
214-
def test_naturaltime_nomonths(test_input: dt.datetime, expected: str) -> None:
216+
def test_naturaltime_nomonths(
217+
test_input: dt.datetime | dt.timedelta | float, expected: str
218+
) -> None:
215219
assert humanize.naturaltime(test_input, months=False) == expected
216220

217221

@@ -506,7 +510,7 @@ def test_naturaltime_timezone_when(test_input: dt.datetime, expected: str) -> No
506510
],
507511
)
508512
def test_precisedelta_one_unit_enough(
509-
val: int | dt.timedelta, min_unit: str, expected: str
513+
val: dt.timedelta | float, min_unit: str, expected: str
510514
) -> None:
511515
assert humanize.precisedelta(val, minimum_unit=min_unit) == expected
512516

@@ -562,7 +566,7 @@ def test_precisedelta_one_unit_enough(
562566
],
563567
)
564568
def test_precisedelta_multiple_units(
565-
val: dt.timedelta, min_unit: str, expected: str
569+
val: dt.timedelta | float, min_unit: str, expected: str
566570
) -> None:
567571
assert humanize.precisedelta(val, minimum_unit=min_unit) == expected
568572

@@ -634,7 +638,7 @@ def test_precisedelta_multiple_units(
634638
],
635639
)
636640
def test_precisedelta_custom_format(
637-
val: dt.timedelta, min_unit: str, fmt: str, expected: str
641+
val: dt.timedelta | float, min_unit: str, fmt: str, expected: str
638642
) -> None:
639643
assert humanize.precisedelta(val, minimum_unit=min_unit, format=fmt) == expected
640644

@@ -711,7 +715,7 @@ def test_precisedelta_custom_format(
711715
],
712716
)
713717
def test_precisedelta_suppress_units(
714-
val: dt.timedelta, min_unit: str, suppress: list[str], expected: str
718+
val: dt.timedelta | float, min_unit: str, suppress: list[str], expected: str
715719
) -> None:
716720
assert (
717721
humanize.precisedelta(val, minimum_unit=min_unit, suppress=suppress) == expected

0 commit comments

Comments
 (0)