temp_commit

dipy · Aug 13, 2018 · e06854a · e06854a
1 parent fe09b3a
commit e06854a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/dipy/workflows/tests/test_vis.py b/dipy/workflows/tests/test_vis.py
@@ -41,7 +41,7 @@ def test_mosaic():
                                      out_dir=temp_out_dir,
                                      out_moved=out_moved,
                                      out_affine=out_affine,
-                                     save_metric=True,
+                                     save_metric=False,
                                      level_iters=[100, 10, 1],
                                      out_quality='affine_q.txt')
 
@@ -54,9 +54,9 @@ def test_mosaic():
 
         x, y, z = overlapped.shape[:3]
 
-        assert overlapped[x//2, :, :, 2] is 0
-        assert overlapped[:, y//2, :, 2] is 0
-        assert overlapped[:, :, z//2, 2] is 0
+        assert overlapped[x//2, :, :, 2] == 0
+        assert overlapped[:, y//2, :, 2] == 0
+        assert overlapped[:, :, z//2, 2] == 0
 
 if __name__ == "__main__":
     run_module_suite()

diff --git a/dipy/workflows/vis_registration.py b/dipy/workflows/vis_registration.py
@@ -139,9 +139,6 @@ def create_mosaic(self, static_img, moved_img, affine, fname):
         overlay, value_range = self.process_image_data(static_img,
                                                        moved_img)
 
-        print(overlay.shape)
-        print(overlay[50,50,50,:])
-
         if fname is None:
             return overlay