Fix test

ACCESS-NRI · Oct 3, 2024 · 2d6bba2 · 2d6bba2
1 parent c5d1a2e
commit 2d6bba2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_translators.py b/tests/test_translators.py
@@ -12,9 +12,9 @@
     DefaultTranslator,
     EraiTranslator,
     TranslatorError,
-    _cmip_frequency_translator,
     _cmip_realm_translator,
     _to_tuple,
+    frequency_translations,
 )
 
 
@@ -67,7 +67,7 @@
 def test_cmip_frequency_translator(input, expected):
     """Test translation of entries in the CMIP frequency column"""
     series = pd.Series(input)
-    translated = _cmip_frequency_translator(series)
+    translated = series.apply(lambda x: frequency_translations.get(x, x))
     assert list(translated) == expected