Add tests to cover function

Author: antonwolfy

tests/skipped_tests.tbl

@@ -245,8 +245,6 @@ tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_interp_inf_to_nan
 tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_heaviside
 tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_heaviside_nan_inf
 
-tests/third_party/cupy/math_tests/test_rounding.py::TestRounding::test_fix
-
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_0_{a_shape=(), b_shape=(), shape=(4, 3, 2)}::test_beta
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_1_{a_shape=(), b_shape=(), shape=(3, 2)}::test_beta
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_2_{a_shape=(), b_shape=(3, 2), shape=(4, 3, 2)}::test_beta

tests/skipped_tests_gpu.tbl

@@ -299,8 +299,6 @@ tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_interp_inf_to_nan
 tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_heaviside
 tests/third_party/cupy/math_tests/test_misc.py::TestMisc::test_heaviside_nan_inf
 
-tests/third_party/cupy/math_tests/test_rounding.py::TestRounding::test_fix
-
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_0_{a_shape=(), b_shape=(), shape=(4, 3, 2)}::test_beta
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_1_{a_shape=(), b_shape=(), shape=(3, 2)}::test_beta
 tests/third_party/cupy/random_tests/test_distributions.py::TestDistributionsBeta_param_2_{a_shape=(), b_shape=(3, 2), shape=(4, 3, 2)}::test_beta

tests/test_mathematical.py

@@ -615,6 +615,73 @@ def test_prepend_append_axis_error(self, xp):
         assert_raises(AxisError, xp.diff, a, axis=3, append=0)
 
 
+class TestFix:
+    @pytest.mark.parametrize(
+        "dt", get_all_dtypes(no_none=True, no_complex=True)
+    )
+    def test_basic(self, dt):
+        a = numpy.array(
+            [[1.0, 1.1, 1.5, 1.8], [-1.0, -1.1, -1.5, -1.8]], dtype=dt
+        )
+        ia = dpnp.array(a)
+
+        result = dpnp.fix(ia)
+        expected = numpy.fix(a)
+        assert_array_equal(result, expected)
+
+    @pytest.mark.parametrize("xp", [numpy, dpnp])
+    @pytest.mark.parametrize("dt", get_complex_dtypes())
+    def test_complex(self, xp, dt):
+        a = xp.array([1.1, -1.1], dtype=dt)
+        with pytest.raises((ValueError, TypeError)):
+            xp.fix(a)
+
+    @pytest.mark.parametrize(
+        "a_dt", get_all_dtypes(no_none=True, no_bool=True, no_complex=True)
+    )
+    def test_out(self, a_dt):
+        a = numpy.array(
+            [[1.0, 1.1, 1.5, 1.8], [-1.0, -1.1, -1.5, -1.8]], dtype=a_dt
+        )
+        ia = dpnp.array(a)
+
+        if a.dtype != numpy.float32 and has_support_aspect64():
+            out_dt = numpy.float64
+        else:
+            out_dt = numpy.float32
+        out = numpy.zeros_like(a, dtype=out_dt)
+        iout = dpnp.array(out)
+
+        result = dpnp.fix(ia, out=iout)
+        expected = numpy.fix(a, out=out)
+        assert_array_equal(result, expected)
+
+    @pytest.mark.skipif(not has_support_aspect16(), reason="no fp16 support")
+    @pytest.mark.parametrize("dt", [bool, numpy.float16])
+    def test_out_float16(self, dt):
+        a = numpy.array(
+            [[1.0, 1.1], [1.5, 1.8], [-1.0, -1.1], [-1.5, -1.8]], dtype=dt
+        )
+        out = numpy.zeros_like(a, dtype=numpy.float16)
+        ia, iout = dpnp.array(a), dpnp.array(out)
+
+        result = dpnp.fix(ia, out=iout)
+        expected = numpy.fix(a, out=out)
+        assert_array_equal(result, expected)
+
+    @pytest.mark.parametrize("xp", [numpy, dpnp])
+    @pytest.mark.parametrize("dt", [bool] + get_integer_dtypes())
+    def test_out_invalid_dtype(self, xp, dt):
+        a = xp.array([[1.5, 1.8], [-1.0, -1.1]])
+        out = xp.zeros_like(a, dtype=dt)
+
+        with pytest.raises((ValueError, TypeError)):
+            xp.fix(a, out=out)
+
+    def test_scalar(self):
+        assert_raises(TypeError, dpnp.fix, -3.4)
+
+
 class TestGradient:
     @pytest.mark.parametrize("dt", get_all_dtypes(no_none=True, no_bool=True))
     def test_basic(self, dt):

tests/test_sycl_queue.py

@@ -422,6 +422,7 @@ def test_meshgrid(device):
         pytest.param("exp2", [0.0, 1.0, 2.0]),
         pytest.param("expm1", [1.0e-10, 1.0, 2.0, 4.0, 7.0]),
         pytest.param("fabs", [-1.2, 1.2]),
+        pytest.param("fix", [2.1, 2.9, -2.1, -2.9]),
         pytest.param("flatnonzero", [-2, -1, 0, 1, 2]),
         pytest.param("floor", [-1.7, -1.5, -0.2, 0.2, 1.5, 1.7, 2.0]),
         pytest.param("gradient", [1.0, 2.0, 4.0, 7.0, 11.0, 16.0]),

tests/test_usm_type.py

@@ -553,6 +553,7 @@ def test_norm(usm_type, ord, axis):
         pytest.param("exp2", [0.0, 1.0, 2.0]),
         pytest.param("expm1", [1.0e-10, 1.0, 2.0, 4.0, 7.0]),
         pytest.param("fabs", [-1.2, 1.2]),
+        pytest.param("fix", [2.1, 2.9, -2.1, -2.9]),
         pytest.param("flatnonzero", [-2, -1, 0, 1, 2]),
         pytest.param("floor", [-1.7, -1.5, -0.2, 0.2, 1.5, 1.7, 2.0]),
         pytest.param("gradient", [1, 2, 4, 7, 11, 16]),