diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2022-01-06 03:26:22 +0100 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2022-01-06 14:41:49 +0100 |
| commit | cc7a8e65de692c6d658b8df7c657b37f36f3d02f (patch) | |
| tree | 21fdc7bdbc63fc331bd8a64dae06f8f7e520e33e /numpy/typing | |
| parent | 7262b9cf9f1998caebdd6da6802e10f2a2d7669a (diff) | |
| download | numpy-cc7a8e65de692c6d658b8df7c657b37f36f3d02f.tar.gz | |
TYP: Allow time manipulation functions to accept `data` and `timedelta` objects
Diffstat (limited to 'numpy/typing')
| -rw-r--r-- | numpy/typing/tests/data/reveal/multiarray.pyi | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/numpy/typing/tests/data/reveal/multiarray.pyi b/numpy/typing/tests/data/reveal/multiarray.pyi index cbe30c520..f8937c379 100644 --- a/numpy/typing/tests/data/reveal/multiarray.pyi +++ b/numpy/typing/tests/data/reveal/multiarray.pyi @@ -1,3 +1,4 @@ +import datetime as dt from typing import Any, TypeVar from pathlib import Path @@ -27,6 +28,10 @@ b_i8_f8_f8 = np.broadcast(AR_i8, AR_f8, AR_f8) nditer_obj: np.nditer +date_scalar: dt.date +date_seq: list[dt.date] +timedelta_seq: list[dt.timedelta] + def func(a: int) -> bool: ... reveal_type(next(b_f8)) # E: tuple[Any] @@ -108,19 +113,26 @@ reveal_type(np.datetime_data(np.dtype(np.timedelta64))) # E: Tuple[builtins.str reveal_type(np.busday_count("2011-01", "2011-02")) # E: {int_} reveal_type(np.busday_count(["2011-01"], "2011-02")) # E: ndarray[Any, dtype[{int_}]] +reveal_type(np.busday_count(["2011-01"], date_scalar)) # E: ndarray[Any, dtype[{int_}]] reveal_type(np.busday_offset(M, m)) # E: datetime64 +reveal_type(np.busday_offset(date_scalar, m)) # E: datetime64 reveal_type(np.busday_offset(M, 5)) # E: datetime64 reveal_type(np.busday_offset(AR_M, m)) # E: ndarray[Any, dtype[datetime64]] +reveal_type(np.busday_offset(M, timedelta_seq)) # E: ndarray[Any, dtype[datetime64]] reveal_type(np.busday_offset("2011-01", "2011-02", roll="forward")) # E: datetime64 reveal_type(np.busday_offset(["2011-01"], "2011-02", roll="forward")) # E: ndarray[Any, dtype[datetime64]] reveal_type(np.is_busday("2012")) # E: bool_ +reveal_type(np.is_busday(date_scalar)) # E: bool_ reveal_type(np.is_busday(["2012"])) # E: ndarray[Any, dtype[bool_]] reveal_type(np.datetime_as_string(M)) # E: str_ reveal_type(np.datetime_as_string(AR_M)) # E: ndarray[Any, dtype[str_]] +reveal_type(np.busdaycalendar(holidays=date_seq)) # E: busdaycalendar +reveal_type(np.busdaycalendar(holidays=[M])) # E: busdaycalendar + reveal_type(np.compare_chararrays("a", "b", "!=", rstrip=False)) # E: ndarray[Any, dtype[bool_]] reveal_type(np.compare_chararrays(b"a", b"a", "==", True)) # E: ndarray[Any, dtype[bool_]] |