transfer variable instead of variable_names to write so that can dist…

…inguish between the two RadiusOfMaximumWinds
noaa-ocs-modeling · Mar 22, 2024 · 58d5606 · 58d5606
1 parent 9a8cdaf
commit 58d5606
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ensembleperturbation/perturbation/atcf.py b/ensembleperturbation/perturbation/atcf.py
@@ -1543,7 +1543,7 @@ def write(
                 write_kwargs = {
                     'filename': output_filename,
                     'perturbation': perturbation_values,
-                    'variables': copy(variable_names),
+                    'variables': copy(variables),
                     'weight': float(perturbation['weights'].values),
                 }
 

diff --git a/tests/test_besttrack_ensemble.py b/tests/test_besttrack_ensemble.py
@@ -10,6 +10,7 @@
     MaximumSustainedWindSpeed,
     perturb_tracks,
     RadiusOfMaximumWinds,
+    RadiusOfMaximumWindsPersistent,
     VortexPerturber,
 )
 from tests import check_reference_directory, DATA_DIRECTORY
@@ -66,7 +67,7 @@ def test_multivariate_besttrack_ensemble():
                 MaximumSustainedWindSpeed: -0.25,
                 CrossTrack: 0.25,
                 'along_track': 0.75,
-                'radius_of_maximum_winds': -1,
+                'radius_of_maximum_winds': -0.5,
             },
             0.75,
         ],
@@ -122,7 +123,7 @@ def test_original_file():
     original_track_0.to_file(output_directory / 'original.22')
 
     gauss_variables = [MaximumSustainedWindSpeed, CrossTrack]
-    range_variables = [RadiusOfMaximumWinds]
+    range_variables = [RadiusOfMaximumWindsPersistent]
 
     perturber = VortexPerturber.from_file(
         output_directory / 'original.22', start_date=datetime(2018, 9, 11), file_deck='b',