Merge pull request #109 from honno/stubbing-n-sig-tests

Remove `generate_stubs.py` + hard-coded stubs, change scope of `test_signatures.py`

Author: honno

.github/workflows/numpy.yml

@@ -43,6 +43,10 @@ jobs:
         # waiting on NumPy to allow/revert distinct NaNs for np.unique
         # https://github.com/numpy/numpy/issues/20326#issuecomment-1012380448
         array_api_tests/test_set_functions.py
+
+        # missing copy arg
+        array_api_tests/test_signatures.py::test_func_signature[reshape]
+
         # https://github.com/numpy/numpy/issues/21211
         array_api_tests/test_special_cases.py::test_iop[__iadd__(x1_i is -0 and x2_i is -0) -> -0]
         # https://github.com/numpy/numpy/issues/21213

array-api

@@ -1,7 +1,7 @@
 import os
 from importlib import import_module
 
-from . import function_stubs
+from . import stubs
 
 # Replace this with a specific array module to test it, for example,
 #
@@ -53,38 +53,18 @@ def __repr__(self):
     __call__ = _raise
     __getattr__ = _raise
 
-_integer_dtypes = [
-    'int8',
-    'int16',
-    'int32',
-    'int64',
-    'uint8',
-    'uint16',
-    'uint32',
-    'uint64',
-]
-
-_floating_dtypes = [
-    'float32',
-    'float64',
-]
-
-_numeric_dtypes = [
-    *_integer_dtypes,
-    *_floating_dtypes,
-]
-
-_boolean_dtypes = [
-    'bool',
-]
-
 _dtypes = [
-    *_boolean_dtypes,
-    *_numeric_dtypes
+    "bool",
+    "uint8", "uint16", "uint32", "uint64",
+    "int8", "int16", "int32", "int64",
+    "float32", "float64",
 ]
+_constants = ["e", "inf", "nan", "pi"]
+_funcs = [f.__name__ for funcs in stubs.category_to_funcs.values() for f in funcs]
+_top_level_attrs = _dtypes + _constants + _funcs + stubs.EXTENSIONS
 
-for func_name in function_stubs.__all__ + _dtypes:
+for attr in _top_level_attrs:
     try:
-        globals()[func_name] = getattr(mod, func_name)
+        globals()[attr] = getattr(mod, attr)
     except AttributeError:
-        globals()[func_name] = _UndefinedStub(func_name)
+        globals()[attr] = _UndefinedStub(attr)