Test fix (#133)

OceanStreamIO · Nov 24, 2023 · 977ff37 · 977ff37
1 parent 00170e4
commit 977ff37
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_shoal_process.py b/tests/test_shoal_process.py
@@ -1,6 +1,7 @@
+from oceanstream.L3_regridded_data.shoal_detection_handler import attach_shoal_mask_to_ds, WEILL_DEFAULT_PARAMETERS
 import pytest
 
-from oceanstream.L3_regridded_data.shoal_detection_handler import attach_shoal_mask_to_ds
+
 from oceanstream.L3_regridded_data.shoal_process import (
     process_shoals,
     process_single_shoal,
@@ -10,7 +11,8 @@
 
 @pytest.mark.ignore
 def prep_dataset(Sv):
-    parameters = {"thr": -55, "maxvgap": -5, "maxhgap": 0, "minvlen": 5, "minhlen": 5}
+    # parameters = {"thr": -55, "maxvgap": -5, "maxhgap": 0, "minvlen": 5, "minhlen": 5}
+    parameters = WEILL_DEFAULT_PARAMETERS
     shoal_dataset = attach_shoal_mask_to_ds(Sv, parameters=parameters, method="will")
     shoal_dataset["mask_shoal"][:, :, 0:25] = False
     shoal_dataset["mask_shoal"][:, :, 600:] = False