add tests

Author: Vahid Tavanashad

dpnp/dpnp_algo/dpnp_elementwise_common.py

@@ -122,9 +122,11 @@ def _call_func(src, dst, sycl_queue, depends=None):
             if vmi._is_available() and mkl_fn_to_call is not None:
                 if getattr(vmi, mkl_fn_to_call)(sycl_queue, src, dst):
                     # call pybind11 extension for unary function from OneMKL VM
+                    print("call vm backend")
                     return getattr(vmi, mkl_impl_fn)(
                         sycl_queue, src, dst, depends
                     )
+            print("call dpctl backend")
             return unary_dp_impl_fn(src, dst, sycl_queue, depends)
 
         super().__init__(
@@ -185,6 +187,7 @@ def __call__(
         if dtype is not None:
             x_usm = dpt.astype(x_usm, dtype, copy=False)
 
+        print("VAHID")
         out_usm = None if out is None else dpnp.get_usm_ndarray(out)
         res_usm = super().__call__(x_usm, out=out_usm, order=order)
 

dpnp/dpnp_utils/dpnp_utils_linearalgebra.py

@@ -332,9 +332,9 @@ def _get_result_shape_vecdot(x1, x2, x1_ndim, x2_ndim):
 
     if x1_ndim == 1 and x2_ndim == 1:
         result_shape = ()
-    elif x1_is_1D and not x2_is_1D:
+    elif x1_ndim == 1:
         result_shape = x2_shape[:-1]
-    elif x2_is_1D and not x1_is_1D:
+    elif x2_ndim == 1:
         result_shape = x1_shape[:-1]
     else:
         if x1_ndim != x2_ndim:

dpnp/tests/helper.py

@@ -36,6 +36,8 @@ def assert_dtype_allclose(
 
     """
 
+    assert dpnp_arr.shape == numpy_arr.shape
+
     list_64bit_types = [numpy.float64, numpy.complex128]
     is_inexact = lambda x: hasattr(x, "dtype") and dpnp.issubdtype(
         x.dtype, dpnp.inexact

dpnp/tests/test_product.py

@@ -1000,6 +1000,8 @@ def setup_method(self):
             ((1, 4, 5), (3, 1, 5)),
             ((1, 1, 4, 5), (3, 1, 5)),
             ((1, 4, 5), (1, 3, 1, 5)),
+            ((2, 1), (1, 1, 1)),
+            ((1, 1, 3), (3,)),
         ],
     )
     def test_basic(self, dtype, shape1, shape2):