Add double type support check condition for test_print_repr

IntelPython · Feb 10, 2023 · 04edcc9 · 04edcc9
1 parent b10be36
commit 04edcc9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dpctl/tests/test_usm_ndarray_print.py b/dpctl/tests/test_usm_ndarray_print.py
@@ -256,7 +256,10 @@ def test_print_repr(self):
         assert repr(x) == "usm_ndarray(0)"
 
         x = dpt.asarray([np.nan, np.inf], sycl_queue=q)
-        assert repr(x) == "usm_ndarray([nan, inf])"
+        if x.sycl_device.has_aspect_fp64:
+            assert repr(x) == "usm_ndarray([nan, inf])"
+        else:
+            assert repr(x) == "usm_ndarray([nan, inf], dtype=float32)"
 
         x = dpt.arange(9, sycl_queue=q, dtype="int64")
         assert repr(x) == "usm_ndarray([0, 1, 2, 3, 4, 5, 6, 7, 8])"