Update tests

Author: dshemetov

jhu/tests/test_run.py

@@ -3,8 +3,8 @@
 from os import listdir
 from os.path import join
 
+import numpy as np
 import pandas as pd
-from delphi_jhu.run import run_module
 
 
 class TestRun:
@@ -48,3 +48,26 @@ def test_output_file_format(self, run_as_module):
             join("receiving", "20200310_state_confirmed_cumulative_num.csv")
         )
         assert (df.columns.values == ["geo_id", "val", "se", "sample_size"]).all()
+
+
+class TestSmooth:
+    def test_output_files_smoothed(self, run_as_module):
+        dates = [str(x) for x in range(20200304, 20200311)]
+
+        smoothed = pd.read_csv(
+            join("receiving", f"{dates[-1]}_state_confirmed_7dav_cumulative_num.csv")
+        )
+
+        raw = pd.concat(
+            [
+                pd.read_csv(
+                    join("receiving", f"{date}_state_confirmed_cumulative_num.csv")
+                )
+                for date in dates
+            ]
+        )
+
+        raw = raw.groupby("geo_id")["val"].mean()
+        df = pd.merge(smoothed, raw, on="geo_id", suffixes=("_smoothed", "_raw"))
+        assert np.allclose(df["val_smoothed"].values, df["val_raw"].values)
+