add tests for negative use cases to improve covergae

Author: vtavana

tests/test_arraycreation.py

@@ -133,19 +133,7 @@ def test_diag_diagflat_raise_error(func):
         getattr(dpnp, func)(ia, k=2.0)
 
 
-<<<<<<< HEAD
 @pytest.mark.parametrize("func", ["diag", "diagflat"])
-=======
-def test_diag_diagflat_raise_error():
-    ia = dpnp.array([0, 1, 2, 3, 4])
-    with pytest.raises(TypeError):
-        dpnp.diag(ia, k=2.0)
-
-    with pytest.raises(TypeError):
-        dpnp.diagflat(ia, k=2.0)
-
-
->>>>>>> add tests for negative use cases to improve covergae
 @pytest.mark.parametrize(
     "seq",
     [
@@ -521,16 +509,6 @@ def test_vander_seq(sequence):
     assert_allclose(vander_func(numpy, sequence), vander_func(dpnp, sequence))
 
 
-@pytest.mark.parametrize(
-    "sequence",
-    [[1, 2, 3, 4], (1, 2, 3, 4)],
-    ids=["[1, 2, 3, 4]", "(1, 2, 3, 4)"],
-)
-def test_vander_seq(sequence):
-    vander_func = lambda xp, x: xp.vander(x)
-    assert_allclose(vander_func(numpy, sequence), vander_func(dpnp, sequence))
-
-
 @pytest.mark.parametrize(
     "shape",
     [(), 0, (0,), (2, 0, 3), (3, 2)],

tests/test_search.py

@@ -1,3 +1,4 @@
+import dpctl.tensor as dpt
 import numpy
 import pytest
 from numpy.testing import assert_allclose
@@ -7,63 +8,54 @@
 from .helper import get_all_dtypes
 
 
+@pytest.mark.parametrize("func", ["argmax", "argmin"])
 @pytest.mark.parametrize("axis", [None, 0, 1, -1, 2, -2])
 @pytest.mark.parametrize("keepdims", [False, True])
 @pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
-def test_argmax_argmin(axis, keepdims, dtype):
+def test_argmax_argmin(func, axis, keepdims, dtype):
     a = numpy.arange(768, dtype=dtype).reshape((4, 4, 6, 8))
     ia = dpnp.array(a)
 
-    np_res = numpy.argmax(a, axis=axis, keepdims=keepdims)
-    dpnp_res = dpnp.argmax(ia, axis=axis, keepdims=keepdims)
-
-    assert dpnp_res.shape == np_res.shape
-    assert_allclose(dpnp_res, np_res)
-
-    np_res = numpy.argmin(a, axis=axis, keepdims=keepdims)
-    dpnp_res = dpnp.argmin(ia, axis=axis, keepdims=keepdims)
+    np_res = getattr(numpy, func)(a, axis=axis, keepdims=keepdims)
+    dpnp_res = getattr(dpnp, func)(ia, axis=axis, keepdims=keepdims)
 
     assert dpnp_res.shape == np_res.shape
     assert_allclose(dpnp_res, np_res)
 
 
+@pytest.mark.parametrize("func", ["argmax", "argmin"])
 @pytest.mark.parametrize("axis", [None, 0, 1, -1])
 @pytest.mark.parametrize("keepdims", [False, True])
-def test_argmax_argmin_bool(axis, keepdims):
+def test_argmax_argmin_bool(func, axis, keepdims):
     a = numpy.arange(2, dtype=dpnp.bool)
     a = numpy.tile(a, (2, 2))
     ia = dpnp.array(a)
 
-    np_res = numpy.argmax(a, axis=axis, keepdims=keepdims)
-    dpnp_res = dpnp.argmax(ia, axis=axis, keepdims=keepdims)
+    np_res = getattr(numpy, func)(a, axis=axis, keepdims=keepdims)
+    dpnp_res = getattr(dpnp, func)(ia, axis=axis, keepdims=keepdims)
 
     assert dpnp_res.shape == np_res.shape
     assert_allclose(dpnp_res, np_res)
 
-    np_res = numpy.argmin(a, axis=axis, keepdims=keepdims)
-    dpnp_res = dpnp.argmin(ia, axis=axis, keepdims=keepdims)
 
-    assert dpnp_res.shape == np_res.shape
-    assert_allclose(dpnp_res, np_res)
-
-
-@pytest.mark.parametrize("axis", [None, 0, 1, -1, 2, -2])
-@pytest.mark.parametrize("keepdims", [False, True])
-@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True))
-def test_argmax_argmin_out(axis, keepdims, dtype):
-    a = numpy.arange(768, dtype=dtype).reshape((4, 4, 6, 8))
+@pytest.mark.parametrize("func", ["argmax", "argmin"])
+def test_argmax_argmin_out(func):
+    a = numpy.arange(6).reshape((2, 3))
     ia = dpnp.array(a)
 
-    np_res = numpy.argmax(a, axis=axis, keepdims=keepdims)
+    np_res = getattr(numpy, func)(a, axis=0)
     dpnp_res = dpnp.array(numpy.empty_like(np_res))
-    dpnp.argmax(ia, axis=axis, keepdims=keepdims, out=dpnp_res)
+    getattr(dpnp, func)(ia, axis=0, out=dpnp_res)
+    assert_allclose(dpnp_res, np_res)
 
-    assert dpnp_res.shape == np_res.shape
+    dpnp_res = dpt.asarray(numpy.empty_like(np_res))
+    getattr(dpnp, func)(ia, axis=0, out=dpnp_res)
     assert_allclose(dpnp_res, np_res)
 
-    np_res = numpy.argmin(a, axis=axis, keepdims=keepdims)
-    dpnp_res = dpnp.array(numpy.empty_like(np_res))
-    dpnp.argmin(ia, axis=axis, keepdims=keepdims, out=dpnp_res)
+    dpnp_res = numpy.empty_like(np_res)
+    with pytest.raises(TypeError):
+        getattr(dpnp, func)(ia, axis=0, out=dpnp_res)
 
-    assert dpnp_res.shape == np_res.shape
-    assert_allclose(dpnp_res, np_res)
+    dpnp_res = dpnp.array(numpy.empty((2, 3)))
+    with pytest.raises(ValueError):
+        getattr(dpnp, func)(ia, axis=0, out=dpnp_res)