Add third party tests for erf function

Author: antonwolfy

dpnp/tests/third_party/cupy/testing/_loops.py

@@ -71,7 +71,7 @@ def _call_func_cupy(impl, args, kw, name, sp_name, scipy_name):
     if sp_name:
         kw[sp_name] = cupyx.scipy.sparse
     if scipy_name:
-        kw[scipy_name] = cupyx.scipy
+        kw[scipy_name] = cupy
     kw[name] = cupy
     result, error = _call_func(impl, args, kw)
     return result, error

dpnp/tests/third_party/cupyx/__init__.py

@@ -0,0 +1,160 @@
+from __future__ import annotations
+
+import unittest
+
+import numpy
+import pytest
+
+import dpnp as cupy
+import dpnp.special
+from dpnp.tests.third_party.cupy import testing
+
+
+def _boundary_inputs(boundary, rtol, atol):
+    left = boundary * (1 - numpy.copysign(rtol, boundary)) - atol
+    right = boundary * (1 + numpy.copysign(rtol, boundary)) + atol
+    return [left, boundary, right]
+
+
+@testing.with_requires("scipy")
+class _TestBase:
+
+    # @testing.with_requires('scipy>=1.16.0')
+    def test_erf(self):
+        self.check_unary("erf")
+
+    @pytest.mark.skip("erfc() is not supported yet")
+    @testing.with_requires("scipy>=1.16.0")
+    def test_erfc(self):
+        self.check_unary("erfc")
+
+    @pytest.mark.skip("erfcx() is not supported yet")
+    @testing.with_requires("scipy>=1.16.0")
+    def test_erfcx(self):
+        self.check_unary("erfcx")
+
+    @pytest.mark.skip("erfinv() is not supported yet")
+    def test_erfinv(self):
+        self.check_unary("erfinv")
+        self.check_unary_random("erfinv", scale=2, offset=-1)
+        self.check_unary_boundary("erfinv", boundary=-1)
+        self.check_unary_boundary("erfinv", boundary=1)
+
+    @pytest.mark.skip("erfcinv() is not supported yet")
+    def test_erfcinv(self):
+        self.check_unary("erfcinv")
+        self.check_unary_random("erfcinv", scale=2, offset=0)
+        self.check_unary_boundary("erfcinv", boundary=0)
+        self.check_unary_boundary("erfcinv", boundary=2)
+
+
+@testing.with_requires("scipy")
+class TestSpecial(unittest.TestCase, _TestBase):
+
+    @testing.for_dtypes(["e", "f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary(self, name, xp, scp, dtype):
+        import scipy.special
+
+        a = testing.shaped_arange((2, 3), xp, dtype)
+        return getattr(scp.special, name)(a)
+
+    @testing.for_dtypes(["f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary_random(self, name, xp, scp, dtype, scale, offset):
+        import scipy.special
+
+        a = testing.shaped_random((2, 3), xp, dtype, scale=scale) + offset
+        return getattr(scp.special, name)(a)
+
+    @testing.for_dtypes(["f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary_boundary(self, name, xp, scp, dtype, boundary):
+        import scipy.special
+
+        a = _boundary_inputs(boundary, 1.0 / 1024, 1.0 / 1024)
+        a = xp.array(a, dtype=dtype)
+        return getattr(scp.special, name)(a)
+
+    @pytest.mark.skip("erfinv() is not supported yet")
+    @testing.with_requires("scipy>=1.4.0")
+    @testing.for_dtypes(["f", "d"])
+    def test_erfinv_behavior(self, dtype):
+        a = cupy.empty((1,), dtype=dtype)
+
+        a[:] = 1.0 + 1e-6
+        a = cupyx.scipy.special.erfinv(a)
+        assert cupy.isnan(a)
+        a[:] = -1.0 - 1e-6
+        a = cupyx.scipy.special.erfinv(a)
+        assert cupy.isnan(a)
+        a[:] = 1.0
+        a = cupyx.scipy.special.erfinv(a)
+        assert numpy.isposinf(cupy.asnumpy(a))
+        a[:] = -1.0
+        a = cupyx.scipy.special.erfinv(a)
+        assert numpy.isneginf(cupy.asnumpy(a))
+
+    @pytest.mark.skip("erfcinv() is not supported yet")
+    @testing.with_requires("scipy>=1.4.0")
+    @testing.for_dtypes(["f", "d"])
+    def test_erfcinv_behavior(self, dtype):
+        a = cupy.empty((1,), dtype=dtype)
+
+        a[:] = 2.0 + 1e-6
+        a = cupyx.scipy.special.erfcinv(a)
+        assert cupy.isnan(a)
+        a[:] = 0.0 - 1e-6
+        a = cupyx.scipy.special.erfcinv(a)
+        assert cupy.isnan(a)
+        a[:] = 0.0
+        a = cupyx.scipy.special.erfcinv(a)
+        assert numpy.isposinf(cupy.asnumpy(a))
+        a[:] = 2.0
+        a = cupyx.scipy.special.erfcinv(a)
+        assert numpy.isneginf(cupy.asnumpy(a))
+
+
+@pytest.mark.skip("fuse() is not supported yet")
+@testing.with_requires("scipy")
+class TestFusionSpecial(unittest.TestCase, _TestBase):
+
+    @testing.for_dtypes(["e", "f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary(self, name, xp, scp, dtype):
+        import scipy.special
+
+        a = testing.shaped_arange((2, 3), xp, dtype)
+
+        @cupy.fuse()
+        def f(x):
+            return getattr(scp.special, name)(x)
+
+        return f(a)
+
+    @testing.for_dtypes(["f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary_random(self, name, xp, scp, dtype, scale, offset):
+        import scipy.special
+
+        a = testing.shaped_random((2, 3), xp, dtype, scale=scale) + offset
+
+        @cupy.fuse()
+        def f(x):
+            return getattr(scp.special, name)(x)
+
+        return f(a)
+
+    @testing.for_dtypes(["f", "d"])
+    @testing.numpy_cupy_allclose(atol=1e-5, scipy_name="scp")
+    def check_unary_boundary(self, name, xp, scp, dtype, boundary):
+        import scipy.special
+
+        a = _boundary_inputs(boundary, 1.0 / 1024, 1.0 / 1024)
+        a = xp.array(a, dtype=dtype)
+
+        @cupy.fuse()
+        def f(x):
+            return getattr(scp.special, name)(x)
+
+        return f(a)