mrCOSTS I/O now allows specifying a file path.

Author: klapo

pydmd/mrcosts.py

@@ -560,7 +560,7 @@ def from_netcdf(self, file_list):
         self._n_data_vars = n_data_vars
         self._n_time_steps = n_time_steps
 
-    def to_netcdf(self, filename):
+    def to_netcdf(self, filename, filepath="."):
         """
         Save the mrCoSTS fit to file in netcdf format.
 
@@ -569,10 +569,14 @@ def to_netcdf(self, filename):
 
         :param filename: Common name shared by each file.
         :type filename: str
+        :param filepath: Path to save the results. Default is the current
+        directory.
+        :type filename: str
+
         """
         for c in self._costs_array:
             c.to_xarray().to_netcdf(
-                ".".join(
+                filepath.join(
                     (
                         filename,
                         f"window={c.window_length:}",

tests/test_mrcosts.py

@@ -292,8 +292,7 @@ def test_netcdf(tmp_path):
     netcdf format and back to mrCOSTS.
     """
     # Move the I/O tests to the temporary test directory.
-    os.chdir(tmp_path)
-    mrc.to_netcdf("tests")
+    mrc.to_netcdf("tests", filepath=tmp_path)
     file_list = glob.glob("*tests*.nc")
     mrc_from_file = mrCOSTS()
     mrc_from_file.from_netcdf(file_list)