TYP: Overhaul the typing test suite

BvB93 · BvB93 · commit fb65dc5198de · 2023-09-04T17:30:02.000+02:00
diff --git a/numpy/typing/mypy_plugin.py b/numpy/typing/mypy_plugin.py
@@ -75,8 +75,7 @@ def _get_precision_dict() -> dict[str, str]:
 
 
 def _get_extended_precision_list() -> list[str]:
-    extended_types = [np.ulonglong, np.longlong, np.longdouble, np.clongdouble]
-    extended_names = {
+    extended_names = [
         "uint128",
         "uint256",
         "int128",
@@ -89,8 +88,8 @@ def _get_extended_precision_list() -> list[str]:
         "complex192",
         "complex256",
         "complex512",
-    }
-    return [i.__name__ for i in extended_types if i.__name__ in extended_names]
+    ]
+    return [i for i in extended_names if hasattr(np, i)]
 
 
 def _get_c_intp_name() -> str:
diff --git a/numpy/typing/tests/data/mypy.ini b/numpy/typing/tests/data/mypy.ini
@@ -2,3 +2,4 @@
 plugins = numpy.typing.mypy_plugin
 show_absolute_path = True
 implicit_reexport = False
+pretty = True
diff --git a/numpy/typing/tests/test_typing.py b/numpy/typing/tests/test_typing.py
