Update unit tests for get_file_suffix()

e-mission · Nov 27, 2024 · 8952810 · 8952810
1 parent 6cca577
commit 8952810
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/viz_scripts/tests/test_scaffolding.py b/viz_scripts/tests/test_scaffolding.py
@@ -189,4 +189,11 @@ def test_get_quality_text_numerator(sensed_df):
 
 def test_get_quality_text_numerator_include_test_users(sensed_df):
     result = scaffolding.get_quality_text_sensed(sensed_df, include_test_users=True)
-    assert result == "Based on 8 trips from 5 testers and participants"
+    assert result == "Based on 8 trips from 5 testers and participants"
+
+def test_get_file_suffix():
+    year = 2024
+    month = 12
+    program = "default"
+    result = scaffolding.get_file_suffix(year, month, program)
+    assert result == "_2024_12_default"