MAINT: add more tests for __array__

Author: person142

tests/fail/array_like.py

@@ -8,3 +8,8 @@ class A:
 x1: "np.ArrayLike" = (i for i in range(10))  # E: Incompatible types in assignment
 x2: "np.ArrayLike" = A()  # E: Incompatible types in assignment
 x3: "np.ArrayLike" = {1: "foo", 2: "bar"}  # E: Incompatible types in assignment
+
+scalar = np.int64(1)
+scalar.__array__(dtype=np.float64)  # E: Unexpected keyword argument
+array = np.array([1])
+array.__array__(dtype=np.float64)  # E: Unexpected keyword argument

tests/pass/array_like.py

@@ -15,14 +15,21 @@
 x11: "np.ArrayLike" = "foo"
 
 
-class B:
+class A:
     def __array__(self, dtype: Optional["np.DtypeLike"] = None) -> np.ndarray:
         return np.array([1, 2, 3])
 
 
-x12: "np.ArrayLike" = B()
-x13: "np._SupportsArray" = np.int64(1)
-x14: "np._SupportsArray" = np.array(1)
+x12: "np.ArrayLike" = A()
+
+scalar: "np._SupportsArray" = np.int64(1)
+scalar.__array__(np.float64)
+array: "np._SupportsArray" = np.array(1)
+array.__array__(np.float64)
+
+a: "np._SupportsArray"
+a.__array__(np.int64)
+a.__array__(dtype=np.int64)
 
 # Escape hatch for when you mean to make something like an object
 # array.