fixed order of replaces and a test

axiom-data-science · Feb 6, 2025 · 82b1809 · 82b1809
1 parent ee8c9c6
commit 82b1809
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/particle_tracking_manager/the_manager.py b/particle_tracking_manager/the_manager.py
@@ -254,7 +254,7 @@ def __init__(
 
         # want output_file to not include any suffix
         output_file = (
-            output_file.replace(".nc", "").replace(".parq", "").replace(".parquet", "")
+            output_file.replace(".nc", "").replace(".parquet", "").replace(".parq", "")
         )
 
         ## set up log for this simulation

diff --git a/tests/test_opendrift.py b/tests/test_opendrift.py
@@ -326,7 +326,7 @@ def test_output_file():
     """make sure output file is parquet if output_format is parquet"""
 
     m = OpenDriftModel(output_format="parquet")
-    assert m.output_file.endswith(".parq")
+    assert m.output_file.endswith(".parquet")
 
     m = OpenDriftModel(output_format="netcdf")
     assert m.output_file.endswith(".nc")