scipy 1.16.2__cp311-cp311-win_arm64.whl

scipy/special/tests/test_spherical_bessel.py

@@ -0,0 +1,400 @@
+#
+# Tests of spherical Bessel functions.
+#
+import numpy as np
+from numpy.testing import (assert_almost_equal, assert_allclose,
+                           assert_array_almost_equal, suppress_warnings)
+import pytest
+from numpy import sin, cos, sinh, cosh, exp, inf, nan, r_, pi
+
+from scipy.special import spherical_jn, spherical_yn, spherical_in, spherical_kn
+from scipy.integrate import quad
+
+
+class TestSphericalJn:
+    def test_spherical_jn_exact(self):
+        # https://dlmf.nist.gov/10.49.E3
+        # Note: exact expression is numerically stable only for small
+        # n or z >> n.
+        x = np.array([0.12, 1.23, 12.34, 123.45, 1234.5])
+        assert_allclose(spherical_jn(2, x),
+                        (-1/x + 3/x**3)*sin(x) - 3/x**2*cos(x))
+
+    def test_spherical_jn_recurrence_complex(self):
+        # https://dlmf.nist.gov/10.51.E1
+        n = np.array([1, 2, 3, 7, 12])
+        x = 1.1 + 1.5j
+        assert_allclose(spherical_jn(n - 1, x) + spherical_jn(n + 1, x),
+                        (2*n + 1)/x*spherical_jn(n, x))
+
+    def test_spherical_jn_recurrence_real(self):
+        # https://dlmf.nist.gov/10.51.E1
+        n = np.array([1, 2, 3, 7, 12])
+        x = 0.12
+        assert_allclose(spherical_jn(n - 1, x) + spherical_jn(n + 1,x),
+                        (2*n + 1)/x*spherical_jn(n, x))
+
+    def test_spherical_jn_inf_real(self):
+        # https://dlmf.nist.gov/10.52.E3
+        n = 6
+        x = np.array([-inf, inf])
+        assert_allclose(spherical_jn(n, x), np.array([0, 0]))
+
+    def test_spherical_jn_inf_complex(self):
+        # https://dlmf.nist.gov/10.52.E3
+        n = 7
+        x = np.array([-inf + 0j, inf + 0j, inf*(1+1j)])
+        with suppress_warnings() as sup:
+            sup.filter(RuntimeWarning, "invalid value encountered in multiply")
+            assert_allclose(spherical_jn(n, x), np.array([0, 0, inf*(1+1j)]))
+
+    def test_spherical_jn_large_arg_1(self):
+        # https://github.com/scipy/scipy/issues/2165
+        # Reference value computed using mpmath, via
+        # besselj(n + mpf(1)/2, z)*sqrt(pi/(2*z))
+        assert_allclose(spherical_jn(2, 3350.507), -0.00029846226538040747)
+
+    def test_spherical_jn_large_arg_2(self):
+        # https://github.com/scipy/scipy/issues/1641
+        # Reference value computed using mpmath, via
+        # besselj(n + mpf(1)/2, z)*sqrt(pi/(2*z))
+        assert_allclose(spherical_jn(2, 10000), 3.0590002633029811e-05)
+
+    def test_spherical_jn_at_zero(self):
+        # https://dlmf.nist.gov/10.52.E1
+        # But note that n = 0 is a special case: j0 = sin(x)/x -> 1
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0
+        assert_allclose(spherical_jn(n, x), np.array([1, 0, 0, 0, 0, 0]))
+
+
+class TestSphericalYn:
+    def test_spherical_yn_exact(self):
+        # https://dlmf.nist.gov/10.49.E5
+        # Note: exact expression is numerically stable only for small
+        # n or z >> n.
+        x = np.array([0.12, 1.23, 12.34, 123.45, 1234.5])
+        assert_allclose(spherical_yn(2, x),
+                        (1/x - 3/x**3)*cos(x) - 3/x**2*sin(x))
+
+    def test_spherical_yn_recurrence_real(self):
+        # https://dlmf.nist.gov/10.51.E1
+        n = np.array([1, 2, 3, 7, 12])
+        x = 0.12
+        assert_allclose(spherical_yn(n - 1, x) + spherical_yn(n + 1,x),
+                        (2*n + 1)/x*spherical_yn(n, x))
+
+    def test_spherical_yn_recurrence_complex(self):
+        # https://dlmf.nist.gov/10.51.E1
+        n = np.array([1, 2, 3, 7, 12])
+        x = 1.1 + 1.5j
+        assert_allclose(spherical_yn(n - 1, x) + spherical_yn(n + 1, x),
+                        (2*n + 1)/x*spherical_yn(n, x))
+
+    def test_spherical_yn_inf_real(self):
+        # https://dlmf.nist.gov/10.52.E3
+        n = 6
+        x = np.array([-inf, inf])
+        assert_allclose(spherical_yn(n, x), np.array([0, 0]))
+
+    def test_spherical_yn_inf_complex(self):
+        # https://dlmf.nist.gov/10.52.E3
+        n = 7
+        x = np.array([-inf + 0j, inf + 0j, inf*(1+1j)])
+        with suppress_warnings() as sup:
+            sup.filter(RuntimeWarning, "invalid value encountered in multiply")
+            assert_allclose(spherical_yn(n, x), np.array([0, 0, inf*(1+1j)]))
+
+    def test_spherical_yn_at_zero(self):
+        # https://dlmf.nist.gov/10.52.E2
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0
+        assert_allclose(spherical_yn(n, x), np.full(n.shape, -inf))
+
+    def test_spherical_yn_at_zero_complex(self):
+        # Consistently with numpy:
+        # >>> -np.cos(0)/0
+        # -inf
+        # >>> -np.cos(0+0j)/(0+0j)
+        # (-inf + nan*j)
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0 + 0j
+        assert_allclose(spherical_yn(n, x), np.full(n.shape, nan))
+
+
+class TestSphericalJnYnCrossProduct:
+    def test_spherical_jn_yn_cross_product_1(self):
+        # https://dlmf.nist.gov/10.50.E3
+        n = np.array([1, 5, 8])
+        x = np.array([0.1, 1, 10])
+        left = (spherical_jn(n + 1, x) * spherical_yn(n, x) -
+                spherical_jn(n, x) * spherical_yn(n + 1, x))
+        right = 1/x**2
+        assert_allclose(left, right)
+
+    def test_spherical_jn_yn_cross_product_2(self):
+        # https://dlmf.nist.gov/10.50.E3
+        n = np.array([1, 5, 8])
+        x = np.array([0.1, 1, 10])
+        left = (spherical_jn(n + 2, x) * spherical_yn(n, x) -
+                spherical_jn(n, x) * spherical_yn(n + 2, x))
+        right = (2*n + 3)/x**3
+        assert_allclose(left, right)
+
+
+class TestSphericalIn:
+    def test_spherical_in_exact(self):
+        # https://dlmf.nist.gov/10.49.E9
+        x = np.array([0.12, 1.23, 12.34, 123.45])
+        assert_allclose(spherical_in(2, x),
+                        (1/x + 3/x**3)*sinh(x) - 3/x**2*cosh(x))
+
+    def test_spherical_in_recurrence_real(self):
+        # https://dlmf.nist.gov/10.51.E4
+        n = np.array([1, 2, 3, 7, 12])
+        x = 0.12
+        assert_allclose(spherical_in(n - 1, x) - spherical_in(n + 1,x),
+                        (2*n + 1)/x*spherical_in(n, x))
+
+    def test_spherical_in_recurrence_complex(self):
+        # https://dlmf.nist.gov/10.51.E1
+        n = np.array([1, 2, 3, 7, 12])
+        x = 1.1 + 1.5j
+        assert_allclose(spherical_in(n - 1, x) - spherical_in(n + 1,x),
+                        (2*n + 1)/x*spherical_in(n, x))
+
+    def test_spherical_in_inf_real(self):
+        # https://dlmf.nist.gov/10.52.E3
+        n = 5
+        x = np.array([-inf, inf])
+        assert_allclose(spherical_in(n, x), np.array([-inf, inf]))
+
+    def test_spherical_in_inf_complex(self):
+        # https://dlmf.nist.gov/10.52.E5
+        # Ideally, i1n(n, 1j*inf) = 0 and i1n(n, (1+1j)*inf) = (1+1j)*inf, but
+        # this appears impossible to achieve because C99 regards any complex
+        # value with at least one infinite  part as a complex infinity, so
+        # 1j*inf cannot be distinguished from (1+1j)*inf.  Therefore, nan is
+        # the correct return value.
+        n = 7
+        x = np.array([-inf + 0j, inf + 0j, inf*(1+1j)])
+        assert_allclose(spherical_in(n, x), np.array([-inf, inf, nan]))
+
+    def test_spherical_in_at_zero(self):
+        # https://dlmf.nist.gov/10.52.E1
+        # But note that n = 0 is a special case: i0 = sinh(x)/x -> 1
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0
+        assert_allclose(spherical_in(n, x), np.array([1, 0, 0, 0, 0, 0]))
+
+
+class TestSphericalKn:
+    def test_spherical_kn_exact(self):
+        # https://dlmf.nist.gov/10.49.E13
+        x = np.array([0.12, 1.23, 12.34, 123.45])
+        assert_allclose(spherical_kn(2, x),
+                        pi/2*exp(-x)*(1/x + 3/x**2 + 3/x**3))
+
+    def test_spherical_kn_recurrence_real(self):
+        # https://dlmf.nist.gov/10.51.E4
+        n = np.array([1, 2, 3, 7, 12])
+        x = 0.12
+        assert_allclose(
+            (-1)**(n - 1)*spherical_kn(n - 1, x) - (-1)**(n + 1)*spherical_kn(n + 1,x),
+            (-1)**n*(2*n + 1)/x*spherical_kn(n, x)
+        )
+
+    def test_spherical_kn_recurrence_complex(self):
+        # https://dlmf.nist.gov/10.51.E4
+        n = np.array([1, 2, 3, 7, 12])
+        x = 1.1 + 1.5j
+        assert_allclose(
+            (-1)**(n - 1)*spherical_kn(n - 1, x) - (-1)**(n + 1)*spherical_kn(n + 1,x),
+            (-1)**n*(2*n + 1)/x*spherical_kn(n, x)
+        )
+
+    def test_spherical_kn_inf_real(self):
+        # https://dlmf.nist.gov/10.52.E6
+        n = 5
+        x = np.array([-inf, inf])
+        assert_allclose(spherical_kn(n, x), np.array([-inf, 0]))
+
+    def test_spherical_kn_inf_complex(self):
+        # https://dlmf.nist.gov/10.52.E6
+        # The behavior at complex infinity depends on the sign of the real
+        # part: if Re(z) >= 0, then the limit is 0; if Re(z) < 0, then it's
+        # z*inf.  This distinction cannot be captured, so we return nan.
+        n = 7
+        x = np.array([-inf + 0j, inf + 0j, inf*(1+1j)])
+        assert_allclose(spherical_kn(n, x), np.array([-inf, 0, nan]))
+
+    def test_spherical_kn_at_zero(self):
+        # https://dlmf.nist.gov/10.52.E2
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0
+        assert_allclose(spherical_kn(n, x), np.full(n.shape, inf))
+
+    def test_spherical_kn_at_zero_complex(self):
+        # https://dlmf.nist.gov/10.52.E2
+        n = np.array([0, 1, 2, 5, 10, 100])
+        x = 0 + 0j
+        assert_allclose(spherical_kn(n, x), np.full(n.shape, nan))
+
+
+class SphericalDerivativesTestCase:
+    def fundamental_theorem(self, n, a, b):
+        integral, tolerance = quad(lambda z: self.df(n, z), a, b)
+        assert_allclose(integral,
+                        self.f(n, b) - self.f(n, a),
+                        atol=tolerance)
+
+    @pytest.mark.slow
+    def test_fundamental_theorem_0(self):
+        self.fundamental_theorem(0, 3.0, 15.0)
+
+    @pytest.mark.slow
+    def test_fundamental_theorem_7(self):
+        self.fundamental_theorem(7, 0.5, 1.2)
+
+
+class TestSphericalJnDerivatives(SphericalDerivativesTestCase):
+    def f(self, n, z):
+        return spherical_jn(n, z)
+
+    def df(self, n, z):
+        return spherical_jn(n, z, derivative=True)
+
+    def test_spherical_jn_d_zero(self):
+        n = np.array([0, 1, 2, 3, 7, 15])
+        assert_allclose(spherical_jn(n, 0, derivative=True),
+                        np.array([0, 1/3, 0, 0, 0, 0]))
+
+
+class TestSphericalYnDerivatives(SphericalDerivativesTestCase):
+    def f(self, n, z):
+        return spherical_yn(n, z)
+
+    def df(self, n, z):
+        return spherical_yn(n, z, derivative=True)
+
+
+class TestSphericalInDerivatives(SphericalDerivativesTestCase):
+    def f(self, n, z):
+        return spherical_in(n, z)
+
+    def df(self, n, z):
+        return spherical_in(n, z, derivative=True)
+
+    def test_spherical_in_d_zero(self):
+        n = np.array([0, 1, 2, 3, 7, 15])
+        spherical_in(n, 0, derivative=False)
+        assert_allclose(spherical_in(n, 0, derivative=True),
+                        np.array([0, 1/3, 0, 0, 0, 0]))
+
+
+class TestSphericalKnDerivatives(SphericalDerivativesTestCase):
+    def f(self, n, z):
+        return spherical_kn(n, z)
+
+    def df(self, n, z):
+        return spherical_kn(n, z, derivative=True)
+
+
+class TestSphericalOld:
+    # These are tests from the TestSpherical class of test_basic.py,
+    # rewritten to use spherical_* instead of sph_* but otherwise unchanged.
+
+    def test_sph_in(self):
+        # This test reproduces test_basic.TestSpherical.test_sph_in.
+        i1n = np.empty((2,2))
+        x = 0.2
+
+        i1n[0][0] = spherical_in(0, x)
+        i1n[0][1] = spherical_in(1, x)
+        i1n[1][0] = spherical_in(0, x, derivative=True)
+        i1n[1][1] = spherical_in(1, x, derivative=True)
+
+        inp0 = (i1n[0][1])
+        inp1 = (i1n[0][0] - 2.0/0.2 * i1n[0][1])
+        assert_array_almost_equal(i1n[0],np.array([1.0066800127054699381,
+                                                0.066933714568029540839]),12)
+        assert_array_almost_equal(i1n[1],[inp0,inp1],12)
+
+    def test_sph_in_kn_order0(self):
+        x = 1.
+        sph_i0 = np.empty((2,))
+        sph_i0[0] = spherical_in(0, x)
+        sph_i0[1] = spherical_in(0, x, derivative=True)
+        sph_i0_expected = np.array([np.sinh(x)/x,
+                                    np.cosh(x)/x-np.sinh(x)/x**2])
+        assert_array_almost_equal(r_[sph_i0], sph_i0_expected)
+
+        sph_k0 = np.empty((2,))
+        sph_k0[0] = spherical_kn(0, x)
+        sph_k0[1] = spherical_kn(0, x, derivative=True)
+        sph_k0_expected = np.array([0.5*pi*exp(-x)/x,
+                                    -0.5*pi*exp(-x)*(1/x+1/x**2)])
+        assert_array_almost_equal(r_[sph_k0], sph_k0_expected)
+
+    def test_sph_jn(self):
+        s1 = np.empty((2,3))
+        x = 0.2
+
+        s1[0][0] = spherical_jn(0, x)
+        s1[0][1] = spherical_jn(1, x)
+        s1[0][2] = spherical_jn(2, x)
+        s1[1][0] = spherical_jn(0, x, derivative=True)
+        s1[1][1] = spherical_jn(1, x, derivative=True)
+        s1[1][2] = spherical_jn(2, x, derivative=True)
+
+        s10 = -s1[0][1]
+        s11 = s1[0][0]-2.0/0.2*s1[0][1]
+        s12 = s1[0][1]-3.0/0.2*s1[0][2]
+        assert_array_almost_equal(s1[0],[0.99334665397530607731,
+                                      0.066400380670322230863,
+                                      0.0026590560795273856680],12)
+        assert_array_almost_equal(s1[1],[s10,s11,s12],12)
+
+    def test_sph_kn(self):
+        kn = np.empty((2,3))
+        x = 0.2
+
+        kn[0][0] = spherical_kn(0, x)
+        kn[0][1] = spherical_kn(1, x)
+        kn[0][2] = spherical_kn(2, x)
+        kn[1][0] = spherical_kn(0, x, derivative=True)
+        kn[1][1] = spherical_kn(1, x, derivative=True)
+        kn[1][2] = spherical_kn(2, x, derivative=True)
+
+        kn0 = -kn[0][1]
+        kn1 = -kn[0][0]-2.0/0.2*kn[0][1]
+        kn2 = -kn[0][1]-3.0/0.2*kn[0][2]
+        assert_array_almost_equal(kn[0],[6.4302962978445670140,
+                                         38.581777787067402086,
+                                         585.15696310385559829],12)
+        assert_array_almost_equal(kn[1],[kn0,kn1,kn2],9)
+
+    def test_sph_yn(self):
+        sy1 = spherical_yn(2, 0.2)
+        sy2 = spherical_yn(0, 0.2)
+        assert_almost_equal(sy1,-377.52483,5)  # previous values in the system
+        assert_almost_equal(sy2,-4.9003329,5)
+        sphpy = (spherical_yn(0, 0.2) - 2*spherical_yn(2, 0.2))/3
+        sy3 = spherical_yn(1, 0.2, derivative=True)
+        # compare correct derivative val. (correct =-system val).
+        assert_almost_equal(sy3,sphpy,4)
+
+
+@pytest.mark.parametrize('derivative', [False, True])
+@pytest.mark.parametrize('fun', [spherical_jn, spherical_in,
+                                 spherical_yn, spherical_kn])
+def test_negative_real_gh14582(derivative, fun):
+    # gh-14582 reported that the spherical Bessel functions did not work
+    # with negative real argument `z`. Check that this is resolved.
+    rng = np.random.default_rng(3598435982345987234)
+    size = 25
+    n = rng.integers(0, 10, size=size)
+    z = rng.standard_normal(size=size)
+    res = fun(n, z, derivative=derivative)
+    ref = fun(n, z+0j, derivative=derivative)
+    np.testing.assert_allclose(res, ref.real)

scipy/special/tests/test_support_alternative_backends.py

@@ -0,0 +1,248 @@
+from functools import partial
+import pickle
+
+import pytest
+from hypothesis import given, strategies
+import hypothesis.extra.numpy as npst
+from packaging import version
+
+from scipy import special
+from scipy.special._support_alternative_backends import _special_funcs
+from scipy._lib._array_api_no_0d import xp_assert_close
+from scipy._lib._array_api import (is_cupy, is_dask, is_jax, is_torch,
+                                   make_xp_pytest_param, make_xp_test_case,
+                                   xp_default_dtype)
+from scipy._lib.array_api_compat import numpy as np
+
+# Run all tests in this module in the Array API CI, including those without
+# the xp fixture
+pytestmark = pytest.mark.array_api_backends
+
+lazy_xp_modules = [special]
+
+
+def _skip_or_tweak_alternative_backends(xp, f_name, dtypes):
+    """Skip tests for specific intersections of scipy.special functions 
+    vs. backends vs. dtypes vs. devices.
+    Also suggest bespoke tweaks.
+
+    Returns
+    -------
+    positive_only : bool
+        Whether you should exclusively test positive inputs.
+    dtypes_np_ref : list[str]
+        The dtypes to use for the reference NumPy arrays.
+    """
+    if ((is_jax(xp) and f_name == 'gammaincc')  # google/jax#20699
+        # gh-20972
+        or ((is_cupy(xp) or is_jax(xp) or is_torch(xp)) and f_name == 'chdtrc')):
+        positive_only = True
+    else:
+        positive_only = False
+
+    if not any('int' in dtype for dtype in dtypes):
+        return positive_only, dtypes
+
+    # Integer-specific issues from this point onwards
+
+    if ((is_torch(xp) and f_name in {'gammainc', 'gammaincc'})
+        or (is_cupy(xp) and f_name in {'stdtr', 'i0e', 'i1e'})
+        or (is_jax(xp) and f_name in {'stdtr', 'ndtr', 'ndtri', 'log_ndtr'})
+    ):
+        pytest.skip(f"`{f_name}` does not support integer types")
+
+    # int/float mismatched args support is sketchy
+    if (any('float' in dtype for dtype in dtypes)
+        and ((is_torch(xp) and f_name in ('rel_entr', 'xlogy'))
+             or (is_jax(xp) and f_name in ('gammainc', 'gammaincc',
+                                           'rel_entr', 'xlogy')))
+    ):
+        pytest.xfail("dtypes do not match")
+
+    dtypes_np_ref = dtypes
+    if (is_torch(xp) and xp_default_dtype(xp) == xp.float32
+        and f_name not in {'betainc', 'betaincc', 'stdtr', 'stdtrit'}
+    ):
+        # On PyTorch with float32 default dtype, sometimes ints are promoted
+        # to float32, and sometimes to float64.
+        # When they are promoted to float32, explicitly convert the reference
+        # numpy arrays to float32 to prevent them from being automatically promoted
+        # to float64 instead.
+        dtypes_np_ref = ['float32' if 'int' in dtype else dtype for dtype in dtypes]
+
+    return positive_only, dtypes_np_ref
+
+
+@pytest.mark.filterwarnings("ignore:invalid value encountered:RuntimeWarning:dask")
+@pytest.mark.parametrize('shapes', [[(0,)]*4, [tuple()]*4, [(10,)]*4,
+                                    [(10,), (11, 1), (12, 1, 1), (13, 1, 1, 1)]])
+@pytest.mark.parametrize('dtype', ['float32', 'float64', 'int64'])
+@pytest.mark.parametrize(
+    'func,nfo', [make_xp_pytest_param(i.wrapper, i) for i in _special_funcs])
+def test_support_alternative_backends(xp, func, nfo, dtype, shapes):
+    positive_only, [dtype_np_ref] = _skip_or_tweak_alternative_backends(
+        xp, nfo.name, [dtype])
+    dtype_np = getattr(np, dtype)
+    dtype_xp = getattr(xp, dtype)
+
+    shapes = shapes[:nfo.n_args]
+    rng = np.random.default_rng(984254252920492019)
+    if 'int' in dtype:
+        iinfo = np.iinfo(dtype_np)
+        rand = partial(rng.integers, iinfo.min, iinfo.max + 1)
+    else:
+        rand = rng.standard_normal
+    args_np = [rand(size=shape, dtype=dtype_np) for shape in shapes]
+    if positive_only:
+        args_np = [np.abs(arg) for arg in args_np]
+
+    args_xp = [xp.asarray(arg, dtype=dtype_xp) for arg in args_np]
+    args_np = [np.asarray(arg, dtype=dtype_np_ref) for arg in args_np]
+
+    if is_dask(xp):
+        # We're using map_blocks to dispatch the function to Dask.
+        # This is the correct thing to do IF all tested functions are elementwise;
+        # otherwise the output would change depending on chunking.
+        # Try to trigger bugs related to having multiple chunks.
+        args_xp = [arg.rechunk(5) for arg in args_xp]
+
+    res = nfo.wrapper(*args_xp)  # Also wrapped by lazy_xp_function
+    ref = nfo.func(*args_np)  # Unwrapped ufunc
+
+    # When dtype_np is integer, the output dtype can be float
+    atol = 0 if ref.dtype.kind in 'iu' else 10 * np.finfo(ref.dtype).eps
+    xp_assert_close(res, xp.asarray(ref), atol=atol)
+
+
+@pytest.mark.parametrize(
+    'func, nfo',
+    [make_xp_pytest_param(i.wrapper, i) for i in _special_funcs if i.n_args >= 2])
+@pytest.mark.filterwarnings("ignore:invalid value encountered:RuntimeWarning:dask")
+def test_support_alternative_backends_mismatched_dtypes(xp, func, nfo):
+    """Test mix-n-match of int and float arguments"""
+    dtypes = ['int64', 'float32', 'float64'][:nfo.n_args]
+    dtypes_xp = [xp.int64, xp.float32, xp.float64][:nfo.n_args]
+    positive_only, dtypes_np_ref = _skip_or_tweak_alternative_backends(
+        xp, nfo.name, dtypes)
+
+    rng = np.random.default_rng(984254252920492019)
+    iinfo = np.iinfo(np.int64)
+    randint = partial(rng.integers, iinfo.min, iinfo.max + 1)
+    args_np = [
+        randint(size=1, dtype=np.int64),
+        rng.standard_normal(size=1, dtype=np.float32),
+        rng.standard_normal(size=1, dtype=np.float64),
+    ][:nfo.n_args]
+    if positive_only:
+        args_np = [np.abs(arg) for arg in args_np]
+
+    args_xp = [xp.asarray(arg, dtype=dtype_xp)
+               for arg, dtype_xp in zip(args_np, dtypes_xp)]
+    args_np = [np.asarray(arg, dtype=dtype_np_ref)
+               for arg, dtype_np_ref in zip(args_np, dtypes_np_ref)]
+
+    res = nfo.wrapper(*args_xp)  # Also wrapped by lazy_xp_function
+    ref = nfo.func(*args_np)  # Unwrapped ufunc
+
+    atol = 10 * np.finfo(ref.dtype).eps
+    xp_assert_close(res, xp.asarray(ref), atol=atol)
+
+
+@pytest.mark.xslow
+@given(data=strategies.data())
+@pytest.mark.fail_slow(5)
+@pytest.mark.parametrize(
+    'func,nfo', [make_xp_pytest_param(nfo.wrapper, nfo) for nfo in _special_funcs])
+@pytest.mark.filterwarnings("ignore:invalid value encountered:RuntimeWarning:dask")
+@pytest.mark.filterwarnings("ignore:divide by zero encountered:RuntimeWarning:dask")
+@pytest.mark.filterwarnings("ignore:overflow encountered:RuntimeWarning:dask")
+@pytest.mark.filterwarnings(
+    "ignore:overflow encountered:RuntimeWarning:array_api_strict"
+)
+def test_support_alternative_backends_hypothesis(xp, func, nfo, data):
+    dtype = data.draw(strategies.sampled_from(['float32', 'float64', 'int64']))
+    positive_only, [dtype_np_ref] = _skip_or_tweak_alternative_backends(
+        xp, nfo.name, [dtype])
+    dtype_np = getattr(np, dtype)
+    dtype_xp = getattr(xp, dtype)
+
+    elements = {'allow_subnormal': False}
+    # Most failures are due to NaN or infinity; uncomment to suppress them
+    # elements['allow_infinity'] = False
+    # elements['allow_nan'] = False
+    if positive_only:
+        elements['min_value'] = 0
+
+    shapes, _ = data.draw(
+        npst.mutually_broadcastable_shapes(num_shapes=nfo.n_args))
+    args_np = [data.draw(npst.arrays(dtype_np, shape, elements=elements))
+               for shape in shapes]
+
+    args_xp = [xp.asarray(arg, dtype=dtype_xp) for arg in args_np]
+    args_np = [np.asarray(arg, dtype=dtype_np_ref) for arg in args_np]
+
+    res = nfo.wrapper(*args_xp)  # Also wrapped by lazy_xp_function
+    ref = nfo.func(*args_np)  # Unwrapped ufunc
+
+    # When dtype_np is integer, the output dtype can be float
+    atol = 0 if ref.dtype.kind in 'iu' else 10 * np.finfo(ref.dtype).eps
+    xp_assert_close(res, xp.asarray(ref), atol=atol)
+
+
+@pytest.mark.parametrize("func", [nfo.wrapper for nfo in _special_funcs])
+def test_pickle(func):
+    roundtrip = pickle.loads(pickle.dumps(func))
+    assert roundtrip is func
+
+
+@pytest.mark.parametrize("func", [nfo.wrapper for nfo in _special_funcs])
+def test_repr(func):
+    assert func.__name__ in repr(func)
+    assert "locals" not in repr(func)
+
+
+@pytest.mark.skipif(
+    version.parse(np.__version__) < version.parse("2.2"),
+    reason="Can't update ufunc __doc__ when SciPy is compiled vs. NumPy < 2.2")
+@pytest.mark.parametrize('func', [nfo.wrapper for nfo in _special_funcs])
+def test_doc(func):
+    """xp_capabilities updates the docstring in place. 
+    Make sure it does so exactly once, including when SCIPY_ARRAY_API is not set.
+    """
+    match = "has experimental support for Python Array API"
+    assert func.__doc__.count(match) == 1
+
+
+@pytest.mark.parametrize('func,n_args',
+                         [(nfo.wrapper, nfo.n_args) for nfo in _special_funcs])
+def test_ufunc_kwargs(func, n_args):
+    """Test that numpy-specific out= and dtype= keyword arguments
+    of ufuncs still work when SCIPY_ARRAY_API is set.
+    """
+    # out=
+    args = [np.asarray([.1, .2])] * n_args
+    out = np.empty(2)
+    y = func(*args, out=out)
+    xp_assert_close(y, out)
+
+    # out= with out.dtype != args.dtype
+    out = np.empty(2, dtype=np.float32)
+    y = func(*args, out=out)
+    xp_assert_close(y, out)
+
+    # dtype=
+    y = func(*args, dtype=np.float32)
+    assert y.dtype == np.float32
+
+
+
+@make_xp_test_case(special.chdtr)
+def test_chdtr_gh21311(xp):
+    # the edge case behavior of generic chdtr was not right; see gh-21311
+    # be sure to test at least these cases
+    # should add `np.nan` into the mix when gh-21317 is resolved
+    x = np.asarray([-np.inf, -1., 0., 1., np.inf])
+    v = x.reshape(-1, 1)
+    ref = special.chdtr(v, x)
+    res = special.chdtr(xp.asarray(v), xp.asarray(x))
+    xp_assert_close(res, xp.asarray(ref))