print arrays when complex special case tests fail

Author: ndgrigorian

.github/workflows/os-llvm-sycl-build.yml

@@ -154,6 +154,18 @@ jobs:
           source set_allvars.sh
           python -m dpctl -f || exit 1
 
+      - name: Run special complex case trig tests
+        shell: bash -l {0}
+        run: |
+          source set_allvars.sh
+          python -m pytest -ra --pyargs dpctl.tests.elementwise.test_trigonometric::test_trig_complex_special_cases -vv || true
+
+      - name: Run special complex case hyperbolic trig tests
+        shell: bash -l {0}
+        run: |
+          source set_allvars.sh
+          python -m pytest -ra --pyargs dpctl.tests.elementwise.test_hyperbolic::test_hyper_complex_special_cases -vv || true
+
       - name: Run dpctl/tests
         shell: bash -l {0}
         env:

dpctl/tests/elementwise/test_hyperbolic.py

@@ -311,5 +311,37 @@ def test_hyper_complex_special_cases(np_call, dpt_call, dtype):
 
     tol = 50 * dpt.finfo(dtype).resolution
     Y = dpt_call(Xc)
-    assert_allclose(dpt.asnumpy(dpt.real(Y)), np.real(Ynp), atol=tol, rtol=tol)
-    assert_allclose(dpt.asnumpy(dpt.imag(Y)), np.imag(Ynp), atol=tol, rtol=tol)
+    Y_x = dpt.real(Y)
+    Y_x_np = dpt.asnumpy(Y_x)
+    Ynp_x = np.real(Ynp)
+    try:
+        assert_allclose(Y_x_np, Ynp_x, rtol=tol)
+    except AssertionError as e:
+        print("Y:")
+        print(Y)
+        print("Y_x:")
+        print(Y_x)
+        print("Y_x_np:")
+        print(Y_x_np)
+        print("Ynp:")
+        print(Ynp)
+        print("Ynp_x:")
+        print(Ynp_x)
+        raise e
+    Y_y = dpt.imag(Y)
+    Y_y_np = dpt.asnumpy(Y_y)
+    Ynp_y = np.imag(Ynp)
+    try:
+        assert_allclose(Y_y_np, Ynp_y, atol=tol, rtol=tol)
+    except AssertionError as e:
+        print("Y:")
+        print(Y)
+        print("Y_y:")
+        print(Y_y)
+        print("Y_y_np:")
+        print(Y_y_np)
+        print("Ynp:")
+        print(Ynp)
+        print("Ynp_y:")
+        print(Ynp_y)
+        raise e

dpctl/tests/elementwise/test_trigonometric.py

@@ -340,5 +340,37 @@ def test_trig_complex_special_cases(np_call, dpt_call, dtype):
 
     tol = 50 * dpt.finfo(dtype).resolution
     Y = dpt_call(Xc)
-    assert_allclose(dpt.asnumpy(dpt.real(Y)), np.real(Ynp), atol=tol, rtol=tol)
-    assert_allclose(dpt.asnumpy(dpt.imag(Y)), np.imag(Ynp), atol=tol, rtol=tol)
+    Y_x = dpt.real(Y)
+    Y_x_np = dpt.asnumpy(Y_x)
+    Ynp_x = np.real(Ynp)
+    try:
+        assert_allclose(Y_x_np, Ynp_x, rtol=tol)
+    except AssertionError as e:
+        print("Y:")
+        print(Y)
+        print("Y_x:")
+        print(Y_x)
+        print("Y_x_np:")
+        print(Y_x_np)
+        print("Ynp:")
+        print(Ynp)
+        print("Ynp_x:")
+        print(Ynp_x)
+        raise e
+    Y_y = dpt.imag(Y)
+    Y_y_np = dpt.asnumpy(Y_y)
+    Ynp_y = np.imag(Ynp)
+    try:
+        assert_allclose(Y_y_np, Ynp_y, atol=tol, rtol=tol)
+    except AssertionError as e:
+        print("Y:")
+        print(Y)
+        print("Y_y:")
+        print(Y_y)
+        print("Y_y_np:")
+        print(Y_y_np)
+        print("Ynp:")
+        print(Ynp)
+        print("Ynp_y:")
+        print(Ynp_y)
+        raise e