Skip to content

Commit d997a0d

Browse files
committed
Fix tests
1 parent b89f0d4 commit d997a0d

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

covid_act_now/tests/test_run.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ def test_output_files(self, clean_receiving_dir):
2121
run_module(self.PARAMS)
2222
csv_files = set(listdir("receiving"))
2323
csv_files.discard(".gitignore")
24+
today = pd.Timestamp.today().date().strftime("%Y%m%d")
2425

2526
expected_files = set()
2627
for signal in SIGNALS:
@@ -30,7 +31,11 @@ def test_output_files(self, clean_receiving_dir):
3031
# All output files exist
3132
assert csv_files == expected_files
3233

34+
expected_columns = [
35+
"geo_id", "val", "se", "sample_size",
36+
"missing_val", "missing_se", "missing_sample_size"
37+
]
3338
# All output files have correct columns
3439
for csv_file in csv_files:
3540
df = pd.read_csv(join("receiving", csv_file))
36-
assert (df.columns.values == ["geo_id", "val", "se", "sample_size"]).all()
41+
assert (df.columns.values == expected_columns).all()

0 commit comments

Comments
 (0)