summaryrefslogtreecommitdiff
path: root/numpy/typing
diff options
context:
space:
mode:
authorMatti Picus <matti.picus@gmail.com>2022-02-24 17:47:59 +0200
committerGitHub <noreply@github.com>2022-02-24 17:47:59 +0200
commitedbdd7ffc7e53807ce24d0708e89c41051621f52 (patch)
tree54e4293d70e8f50fc317841b9bdfc05ce9d1c3c5 /numpy/typing
parent73e48ae177c4b29cdbdbcfd39e22cf5de6c7f125 (diff)
parent370b404d9652ab1f4086f2726ed63a9e0e013029 (diff)
downloadnumpy-edbdd7ffc7e53807ce24d0708e89c41051621f52.tar.gz
Merge pull request #21111 from BvB93/__array_finalize__
TYP, MAINT: Relax the `obj` type in `__array_finalize__`
Diffstat (limited to 'numpy/typing')
-rw-r--r--numpy/typing/tests/data/fail/ndarray_misc.pyi2
-rw-r--r--numpy/typing/tests/data/reveal/chararray.pyi3
-rw-r--r--numpy/typing/tests/data/reveal/memmap.pyi2
-rw-r--r--numpy/typing/tests/data/reveal/ndarray_misc.pyi4
-rw-r--r--numpy/typing/tests/data/reveal/rec.pyi1
5 files changed, 12 insertions, 0 deletions
diff --git a/numpy/typing/tests/data/fail/ndarray_misc.pyi b/numpy/typing/tests/data/fail/ndarray_misc.pyi
index 8320a44f3..77bd9a44e 100644
--- a/numpy/typing/tests/data/fail/ndarray_misc.pyi
+++ b/numpy/typing/tests/data/fail/ndarray_misc.pyi
@@ -39,3 +39,5 @@ AR_b.__index__() # E: Invalid self argument
AR_f8[1.5] # E: No overload variant
AR_f8["field_a"] # E: No overload variant
AR_f8[["field_a", "field_b"]] # E: Invalid index type
+
+AR_f8.__array_finalize__(object()) # E: incompatible type
diff --git a/numpy/typing/tests/data/reveal/chararray.pyi b/numpy/typing/tests/data/reveal/chararray.pyi
index 3da2e1599..61906c860 100644
--- a/numpy/typing/tests/data/reveal/chararray.pyi
+++ b/numpy/typing/tests/data/reveal/chararray.pyi
@@ -127,3 +127,6 @@ reveal_type(AR_S.istitle()) # E: ndarray[Any, dtype[bool_]]
reveal_type(AR_U.isupper()) # E: ndarray[Any, dtype[bool_]]
reveal_type(AR_S.isupper()) # E: ndarray[Any, dtype[bool_]]
+
+reveal_type(AR_U.__array_finalize__(object())) # E: None
+reveal_type(AR_S.__array_finalize__(object())) # E: None
diff --git a/numpy/typing/tests/data/reveal/memmap.pyi b/numpy/typing/tests/data/reveal/memmap.pyi
index 86de8eb08..af7307499 100644
--- a/numpy/typing/tests/data/reveal/memmap.pyi
+++ b/numpy/typing/tests/data/reveal/memmap.pyi
@@ -14,3 +14,5 @@ reveal_type(np.memmap("file.txt", offset=5)) # E: memmap[Any, dtype[{uint8}]]
reveal_type(np.memmap(b"file.txt", dtype=np.float64, shape=(10, 3))) # E: memmap[Any, dtype[{float64}]]
with open("file.txt", "rb") as f:
reveal_type(np.memmap(f, dtype=float, order="K")) # E: memmap[Any, dtype[Any]]
+
+reveal_type(memmap_obj.__array_finalize__(object())) # E: None
diff --git a/numpy/typing/tests/data/reveal/ndarray_misc.pyi b/numpy/typing/tests/data/reveal/ndarray_misc.pyi
index f91d6351b..c9a42b3e8 100644
--- a/numpy/typing/tests/data/reveal/ndarray_misc.pyi
+++ b/numpy/typing/tests/data/reveal/ndarray_misc.pyi
@@ -212,3 +212,7 @@ reveal_type(AR_f8.dump("test_file")) # E: None
reveal_type(AR_f8.dump(b"test_file")) # E: None
with open("test_file", "wb") as f:
reveal_type(AR_f8.dump(f)) # E: None
+
+reveal_type(AR_f8.__array_finalize__(None)) # E: None
+reveal_type(AR_f8.__array_finalize__(B)) # E: None
+reveal_type(AR_f8.__array_finalize__(AR_f8)) # E: None
diff --git a/numpy/typing/tests/data/reveal/rec.pyi b/numpy/typing/tests/data/reveal/rec.pyi
index b2eaca899..8ea4a6ee8 100644
--- a/numpy/typing/tests/data/reveal/rec.pyi
+++ b/numpy/typing/tests/data/reveal/rec.pyi
@@ -33,6 +33,7 @@ reveal_type(REC_AR_V.field(0, AR_i8)) # E: None
reveal_type(REC_AR_V.field("field_a", AR_i8)) # E: None
reveal_type(REC_AR_V["field_a"]) # E: Any
reveal_type(REC_AR_V.field_a) # E: Any
+reveal_type(REC_AR_V.__array_finalize__(object())) # E: None
reveal_type(np.recarray( # recarray[Any, dtype[record]]
shape=(10, 5),