Adds new tests for reductions

Author: ndgrigorian

dpctl/tests/test_tensor_sum.py

@@ -173,6 +173,21 @@ def test_largish_reduction(arg_dtype, n):
     assert dpt.all(dpt.equal(y1, n * m))
 
 
+@pytest.mark.parametrize("n", [1023, 1024, 1025])
+def test_largish_reduction_axis1_axis0(n):
+    get_queue_or_skip()
+
+    m = 25
+    x1 = dpt.ones((m, n), dtype="f4")
+    x2 = dpt.ones((n, m), dtype="f4")
+
+    y1 = dpt.sum(x1, axis=1)
+    y2 = dpt.sum(x2, axis=0)
+
+    assert dpt.all(y1 == n)
+    assert dpt.all(y2 == n)
+
+
 def test_axis0_bug():
     "gh-1391"
     get_queue_or_skip()

dpctl/tests/test_usm_ndarray_reductions.py

@@ -326,6 +326,28 @@ def test_logsumexp_keepdims():
     assert s.shape == (3, 1, 1, 6, 1)
 
 
+def test_logsumexp_keepdims_zero_size():
+    get_queue_or_skip()
+    n = 10
+    a = dpt.ones((n, 0, n))
+
+    s1 = dpt.logsumexp(a, keepdims=True)
+    assert s1.shape == (1, 1, 1)
+
+    s2 = dpt.logsumexp(a, axis=(0, 1), keepdims=True)
+    assert s2.shape == (1, 1, n)
+
+    s3 = dpt.logsumexp(a, axis=(1, 2), keepdims=True)
+    assert s3.shape == (n, 1, 1)
+
+    s4 = dpt.logsumexp(a, axis=(0, 2), keepdims=True)
+    assert s4.shape == (1, 0, 1)
+
+    a0 = a[0]
+    s5 = dpt.logsumexp(a0, keepdims=True)
+    assert s5.shape == (1, 1)
+
+
 def test_logsumexp_scalar():
     get_queue_or_skip()
 
@@ -337,6 +359,29 @@ def test_logsumexp_scalar():
     assert s.shape == ()
 
 
+def test_logsumexp_complex():
+    get_queue_or_skip()
+
+    x = dpt.zeros(1, dtype="c8")
+    with pytest.raises(TypeError):
+        dpt.logsumexp(x)
+
+
+def test_logsumexp_int_axis():
+    get_queue_or_skip()
+
+    x = dpt.zeros((8, 10), dtype="f4")
+    res = dpt.logsumexp(x, axis=0)
+    assert res.ndim == 1
+    assert res.shape[0] == 10
+
+
+def test_logsumexp_invalid_arr():
+    x = dict()
+    with pytest.raises(TypeError):
+        dpt.logsumexp(x)
+
+
 @pytest.mark.parametrize("arg_dtype", _no_complex_dtypes[1:])
 def test_hypot_arg_dtype_default_output_dtype_matrix(arg_dtype):
     q = get_queue_or_skip()
@@ -376,3 +421,11 @@ def test_hypot_arg_out_dtype_matrix(arg_dtype, out_dtype):
 
     assert isinstance(r, dpt.usm_ndarray)
     assert r.dtype == dpt.dtype(out_dtype)
+
+
+def test_hypot_complex():
+    get_queue_or_skip()
+
+    x = dpt.zeros(1, dtype="c8")
+    with pytest.raises(TypeError):
+        dpt.reduce_hypot(x)