skip split mismatch test on 1 process

helmholtz-analytics · Jun 19, 2023 · 17ca24f · 17ca24f
1 parent 49197b3
commit 17ca24f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/heat/core/tests/test_operations.py b/heat/core/tests/test_operations.py
@@ -96,13 +96,13 @@ def test___binary_bit_op_broadcast(self):
             c = ht.array([1, 2, 3, 4], comm=ht.MPI_SELF)
             with self.assertRaises(NotImplementedError):
                 b + c
+            with self.assertRaises(NotImplementedError):
+                a.resplit(1) * b
         # skip tests on arm64 architecture
         if platform.machine() != "arm64":
             with self.assertRaises(TypeError):
                 ht.minimum(a, np.float128(1))
             with self.assertRaises(TypeError):
                 ht.minimum(np.float128(1), a)
-        with self.assertRaises(NotImplementedError):
-            a.resplit(1) * b
         with self.assertRaises(ValueError):
             a[2:] * b