turn on usafacts::test_run

Author: sgsmob

usafacts/tests/test_run.py

@@ -1,5 +1,6 @@
 import pytest
 
+from itertools import product
 from os import listdir
 from os.path import join
 
@@ -10,7 +11,7 @@
 class TestRun:
     def test_output_files_exist(self, run_as_module):
 
-        csv_files = listdir("receiving")
+        csv_files = [f for f in listdir("receiving") if f.endswith(".csv")]
 
         dates = [
             "20200229",
@@ -23,27 +24,25 @@ def test_output_files_exist(self, run_as_module):
             "20200307",
             "20200308",
             "20200309",
-            "202003010",
+            "20200310",
         ]
         geos = ["county", "hrr", "msa", "state"]
-        metrics = [
-            "deaths_cumulative_num",
-            "deaths_incidence_num",
-            "deaths_incidence_prop",
-            "confirmed_cumulative_num",
-            "confirmed_incidence_num",
-            "confirmed_incidence_prop",
-            "deaths_7dav_cumulative_prop",
-            "confirmed_7dav_cumulative_prop",
-        ]
+
+        # enumerate metric names.
+        metrics = []
+        for event, span, stat in product(["deaths", "confirmed"],
+                                         ["cumulative", "incidence"],
+                                         ["num", "prop"]):
+            metrics.append("_".join([event, span, stat]))
+            metrics.append("_".join([event, "7dav", span, stat]))
 
         expected_files = []
         for date in dates:
             for geo in geos:
                 for metric in metrics:
                     expected_files += [date + "_" + geo + "_" + metric + ".csv"]
 
-        set(csv_files) == set(expected_files)
+        assert set(csv_files) == set(expected_files)
 
     def test_output_file_format(self, run_as_module):