Rebasing

Author: Sparks29032

tests/test_morphio.py

@@ -329,6 +329,78 @@ def test_grid_selection(self, setup, tmp_path):
                     expected = filter(ignore_path, tf)
                     are_files_same(actual, expected)
 
+    def test_grid_selection(self, setup, tmp_path):
+        morph_file = self.testfiles[0]
+
+        # Save on default grid
+        tmp_default = tmp_path.joinpath("default")
+        tmp_default_name = tmp_default.resolve().as_posix()
+
+        opts, pargs = self.parser.parse_args(
+            [
+                "--multiple-morphs",
+                "-s",
+                tmp_default_name,
+                "-n",
+                "--save-names-file",
+                tssf,
+                "--xmax",
+                "35",
+            ]
+        )
+
+        pargs = [testsequence_dir, morph_file]
+        multiple_morphs(self.parser, opts, pargs, stdout_flag=False)
+
+        tmp_default_morphs = tmp_default.joinpath("Morphs")
+
+        # Check the saved files are the same for verbose
+        common = []
+        for item in tmp_default_morphs.glob("**/*.*"):
+            if item.is_file():
+                common.append(item.relative_to(tmp_default_morphs).as_posix())
+        for file in common:
+            with open(tmp_default_morphs.joinpath(file)) as gf:
+                with open(test_saving_grid_default.joinpath(file)) as tf:
+                    actual = filter(ignore_path, gf)
+                    expected = filter(ignore_path, tf)
+                    are_files_same(actual, expected)
+
+        # Save on original grid
+        tmp_original = tmp_path.joinpath("original")
+        tmp_original_name = tmp_original.resolve().as_posix()
+
+        opts, pargs = self.parser.parse_args(
+            [
+                "--multiple-morphs",
+                "-s",
+                tmp_original_name,
+                "-n",
+                "--save-names-file",
+                tssf,
+                "--xmax",
+                "35",
+                "-o",
+            ]
+        )
+
+        pargs = [testsequence_dir, morph_file]
+        multiple_morphs(self.parser, opts, pargs, stdout_flag=False)
+
+        tmp_original_morphs = tmp_default.joinpath("Morphs")
+
+        # Check the saved files are the same for verbose
+        common = []
+        for item in tmp_original_morphs.glob("**/*.*"):
+            if item.is_file():
+                common.append(item.relative_to(tmp_original_morphs).as_posix())
+        for file in common:
+            with open(tmp_original_morphs.joinpath(file)) as gf:
+                with open(test_saving_grid_original.joinpath(file)) as tf:
+                    actual = filter(ignore_path, gf)
+                    expected = filter(ignore_path, tf)
+                    are_files_same(actual, expected)
+
     # Similar format as test_morph_outputs
     def test_morph_diff_outputs(self, setup, tmp_path):
         morph_file = self.testfiles[0]