new raise test now only runs on multiple processes

helmholtz-analytics · Jul 20, 2021 · 89fb977 · 89fb977
1 parent bf39b25
commit 89fb977
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/heat/core/tests/test_dndarray.py b/heat/core/tests/test_dndarray.py
@@ -1333,10 +1333,11 @@ def test_setitem_getitem(self):
             a[..., ...]
         with self.assertRaises(ValueError):
             a[..., ...] = 1
-        with self.assertRaises(ValueError):
-            x = ht.ones((10, 10), split=0)
-            setting = ht.zeros((8, 8), split=1)
-            x[1:-1, 1:-1] = setting
+        if a.comm.size > 1:
+            with self.assertRaises(ValueError):
+                x = ht.ones((10, 10), split=0)
+                setting = ht.zeros((8, 8), split=1)
+                x[1:-1, 1:-1] = setting
 
     def test_size_gnumel(self):
         a = ht.zeros((10, 10, 10), split=None)