TST: Update the typing tests for np.number subclasses

Author: Bas van Beek

numpy/typing/tests/data/fail/scalars.py

@@ -1,7 +1,9 @@
+import sys
 import numpy as np
 
 f2: np.float16
 f8: np.float64
+c8: np.complex64
 
 # Construction
 
@@ -80,3 +82,13 @@ def func(a: np.float32) -> None: ...
 
 func(f2)  # E: incompatible type
 func(f8)  # E: incompatible type
+
+round(c8)  # E: No overload variant
+
+c8.__getnewargs__()  # E: Invalid self argument
+f2.__getnewargs__()  # E: Invalid self argument
+f2.is_integer()  # E: Invalid self argument
+f2.hex()  # E: Invalid self argument
+np.float16.fromhex("0x0.0p+0")  # E: Invalid self argument
+f2.__trunc__()  # E: Invalid self argument
+f2.__getformat__("float")  # E: Invalid self argument

numpy/typing/tests/data/reveal/scalars.py

@@ -1,3 +1,4 @@
+import sys
 import numpy as np
 
 b: np.bool_
@@ -6,6 +7,7 @@
 f8: np.float64
 c8: np.complex64
 c16: np.complex128
+m: np.timedelta64
 U: np.str_
 S: np.bytes_
 
@@ -126,3 +128,31 @@
 reveal_type(i8.getfield(float))  # E: Any
 reveal_type(i8.getfield(np.float64))  # E: {float64}
 reveal_type(i8.getfield(np.float64, 8))  # E: {float64}
+
+reveal_type(f8.as_integer_ratio())  # E: Tuple[builtins.int, builtins.int]
+reveal_type(f8.is_integer())  # E: bool
+reveal_type(f8.__trunc__())  # E: int
+reveal_type(f8.__getformat__("float"))  # E: str
+reveal_type(f8.hex())  # E: str
+reveal_type(np.float64.fromhex("0x0.0p+0"))  # E: {float64}
+
+reveal_type(f8.__getnewargs__())  # E: Tuple[builtins.float]
+reveal_type(c16.__getnewargs__())  # E: Tuple[builtins.float, builtins.float]
+
+reveal_type(i8.numerator)  # E: {int64}
+reveal_type(i8.denominator)  # E: Literal[1]
+reveal_type(u8.numerator)  # E: {uint64}
+reveal_type(u8.denominator)  # E: Literal[1]
+reveal_type(m.numerator)  # E: numpy.timedelta64
+reveal_type(m.denominator)  # E: Literal[1]
+
+reveal_type(round(i8))  # E: int
+reveal_type(round(i8, 3))  # E: {int64}
+reveal_type(round(u8))  # E: int
+reveal_type(round(u8, 3))  # E: {uint64}
+reveal_type(round(f8))  # E: int
+reveal_type(round(f8, 3))  # E: {float64}
+
+if sys.version_info >= (3, 9):
+    reveal_type(f8.__ceil__())  # E: int
+    reveal_type(f8.__floor__())  # E: int