diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2021-02-16 14:51:49 +0100 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2021-02-25 14:06:37 +0100 |
| commit | f345d732a97b647de1fb26aeae533ca48e8229e9 (patch) | |
| tree | 8fa40fd5427def493badf6495789b4b671db2d6b /numpy/typing | |
| parent | 8a18515f6b5eb97a184b92858741eb4a5aab613a (diff) | |
| download | numpy-f345d732a97b647de1fb26aeae533ca48e8229e9.tar.gz | |
MAINT: Relax the type-constraints of `IndexExpression.__getitem__`
Diffstat (limited to 'numpy/typing')
| -rw-r--r-- | numpy/typing/tests/data/fail/index_tricks.py | 4 | ||||
| -rw-r--r-- | numpy/typing/tests/data/pass/index_tricks.py | 2 | ||||
| -rw-r--r-- | numpy/typing/tests/data/reveal/index_tricks.py | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/numpy/typing/tests/data/fail/index_tricks.py b/numpy/typing/tests/data/fail/index_tricks.py index 706f135b2..cbc43fd54 100644 --- a/numpy/typing/tests/data/fail/index_tricks.py +++ b/numpy/typing/tests/data/fail/index_tricks.py @@ -10,9 +10,5 @@ np.mgrid[1] # E: Invalid index type np.mgrid[...] # E: Invalid index type np.ogrid[1] # E: Invalid index type np.ogrid[...] # E: Invalid index type -np.index_exp[1] # E: No overload variant -np.index_exp[...] # E: No overload variant -np.s_[1] # E: cannot be "int" -np.s_[...] # E: cannot be "ellipsis" np.fill_diagonal(AR_LIKE_f, 2) # E: incompatible type np.diag_indices(1.0) # E: incompatible type diff --git a/numpy/typing/tests/data/pass/index_tricks.py b/numpy/typing/tests/data/pass/index_tricks.py index ce7f415f3..4c4c11959 100644 --- a/numpy/typing/tests/data/pass/index_tricks.py +++ b/numpy/typing/tests/data/pass/index_tricks.py @@ -46,9 +46,11 @@ np.ogrid[1:1:2, None:10] np.index_exp[0:1] np.index_exp[0:1, None:3] +np.index_exp[0, 0:1, ..., [0, 1, 3]] np.s_[0:1] np.s_[0:1, None:3] +np.s_[0, 0:1, ..., [0, 1, 3]] np.ix_(AR_LIKE_b[0]) np.ix_(AR_LIKE_i[0], AR_LIKE_f[0]) diff --git a/numpy/typing/tests/data/reveal/index_tricks.py b/numpy/typing/tests/data/reveal/index_tricks.py index dc061d314..ec2013025 100644 --- a/numpy/typing/tests/data/reveal/index_tricks.py +++ b/numpy/typing/tests/data/reveal/index_tricks.py @@ -45,9 +45,11 @@ reveal_type(np.ogrid[1:1:2, None:10]) # E: list[numpy.ndarray[Any, numpy.dtype[ reveal_type(np.index_exp[0:1]) # E: Tuple[builtins.slice] reveal_type(np.index_exp[0:1, None:3]) # E: Tuple[builtins.slice, builtins.slice] +reveal_type(np.index_exp[0, 0:1, ..., [0, 1, 3]]) # E: Tuple[Literal[0]?, builtins.slice, builtins.ellipsis, builtins.list[builtins.int]] reveal_type(np.s_[0:1]) # E: builtins.slice reveal_type(np.s_[0:1, None:3]) # E: Tuple[builtins.slice, builtins.slice] +reveal_type(np.s_[0, 0:1, ..., [0, 1, 3]]) # E: Tuple[Literal[0]?, builtins.slice, builtins.ellipsis, builtins.list[builtins.int]] reveal_type(np.ix_(AR_LIKE_b)) # E: tuple[numpy.ndarray[Any, numpy.dtype[numpy.bool_]]] reveal_type(np.ix_(AR_LIKE_i, AR_LIKE_f)) # E: tuple[numpy.ndarray[Any, numpy.dtype[{double}]]] |