Add a draft reciprocal function for 2024.12

Author: asmeurer

array_api_strict/__init__.py

@@ -177,6 +177,7 @@
     positive,
     pow,
     real,
+    reciprocal,
     remainder,
     round,
     sign,
@@ -246,6 +247,7 @@
     "positive",
     "pow",
     "real",
+    "reciprocal",
     "remainder",
     "round",
     "sign",

array_api_strict/_elementwise_functions.py

@@ -872,6 +872,17 @@ def real(x: Array, /) -> Array:
     return Array._new(np.real(x._array), device=x.device)
 
 
+@requires_api_version('2024.12')
+def reciprocal(x: Array, /) -> Array:
+    """
+    Array API compatible wrapper for :py:func:`np.reciprocal <numpy.reciprocal>`.
+
+    See its docstring for more information.
+    """
+    if x.dtype not in _floating_dtypes:
+        raise TypeError("Only floating-point dtypes are allowed in reciprocal")
+    return Array._new(np.reciprocal(x._array), device=x.device)
+
 def remainder(x1: Array, x2: Array, /) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.remainder <numpy.remainder>`.

array_api_strict/tests/test_elementwise_functions.py

@@ -86,6 +86,7 @@ def nargs(func):
     "positive": "numeric",
     "pow": "numeric",
     "real": "complex floating-point",
+    "reciprocal": "floating-point",
     "remainder": "real numeric",
     "round": "numeric",
     "sign": "numeric",

array_api_strict/tests/test_flags.py

@@ -287,6 +287,7 @@ def test_fft(func_name):
 api_version_2024_12_examples = {
     'diff': lambda: xp.diff(xp.asarray([0, 1, 2])),
     'nextafter': lambda: xp.nextafter(xp.asarray(0.), xp.asarray(1.)),
+    'reciprocal': lambda: xp.reciprocal(xp.asarray([2.])),
 }
 
 @pytest.mark.parametrize('func_name', api_version_2023_12_examples.keys())