fix assert array equal

mjirik · Oct 20, 2018 · d089401 · d089401
1 parent e69914e
commit d089401
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/image_manipulation_test.py b/tests/image_manipulation_test.py
@@ -477,8 +477,8 @@ def test_biggest_object(self):
         seg_biggest = ima.get_one_biggest_object(segmentation)
         # newlab = ima.get_nlabels(datap["slab"], "new", return_mode="str")
         self.assertEqual(type(seg_biggest), np.ndarray)
-        self.assertEqual(np.array_equal(seg_biggest.shape), segmentation.shape)
-        self.assertEqual(np.array_equal(seg_biggest.shape), segmentation.shape)
+        self.assertTrue(np.array_equal(seg_biggest.shape, segmentation.shape))
+        self.assertTrue(np.array_equal(seg_biggest.shape, segmentation.shape))
 
     def test_biggest_object_label(self):
         datap = io3d.datasets.generate_abdominal()