Test info functions in the signature tests

Author: asmeurer

array_api_tests/test_signatures.py

@@ -31,7 +31,8 @@ def squeeze(x, /, axis):
 
 from . import dtype_helpers as dh
 from . import xp
-from .stubs import array_methods, category_to_funcs, extension_to_funcs, name_to_func
+from .stubs import (array_methods, category_to_funcs, extension_to_funcs,
+                    name_to_func, info_funcs)
 
 ParameterKind = Literal[
     Parameter.POSITIONAL_ONLY,
@@ -307,3 +308,15 @@ def test_array_method_signature(stub: FunctionType):
     assert hasattr(x, stub.__name__), f"{stub.__name__} not found in array object {x!r}"
     method = getattr(x, stub.__name__)
     _test_func_signature(method, stub, is_method=True)
+
+if info_funcs: # pytest fails collecting if info_funcs is empty
+    @pytest.mark.min_version("2023.12")
+    @pytest.mark.parametrize("stub", info_funcs, ids=lambda f: f.__name__)
+    def test_info_func_signature(stub: FunctionType):
+        try:
+            info_namespace = xp.__array_namespace_info__()
+        except Exception as e:
+            raise AssertionError(f"Could not get info namespace from xp.__array_namespace_info__(): {e}")
+
+        func = getattr(info_namespace, stub.__name__)
+        _test_func_signature(func, stub)