Add signbit

Author: asmeurer

array_api_strict/__init__.py

@@ -177,6 +177,7 @@
     remainder,
     round,
     sign,
+    signbit,
     sin,
     sinh,
     square,
@@ -244,6 +245,7 @@
     "remainder",
     "round",
     "sign",
+    "signbit",
     "sin",
     "sinh",
     "square",

array_api_strict/_elementwise_functions.py

@@ -791,6 +791,18 @@ def sign(x: Array, /) -> Array:
     return Array._new(np.sign(x._array))
 
 
+@requires_api_version('2023.12')
+def signbit(x: Array, /) -> Array:
+    """
+    Array API compatible wrapper for :py:func:`np.signbit <numpy.signbit>`.
+
+    See its docstring for more information.
+    """
+    if x.dtype not in _real_floating_dtypes:
+        raise TypeError("Only real floating-point dtypes are allowed in signbit")
+    return Array._new(np.signbit(x._array))
+
+
 def sin(x: Array, /) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.sin <numpy.sin>`.

array_api_strict/tests/test_elementwise_functions.py

@@ -75,6 +75,7 @@ def nargs(func):
     "remainder": "real numeric",
     "round": "numeric",
     "sign": "numeric",
+    "signbit": "real floating-point",
     "sin": "floating-point",
     "sinh": "floating-point",
     "sqrt": "floating-point",