logging for visiblity for debug

cmu-delphi · Jul 24, 2024 · f59b925 · f59b925
1 parent b35514a
commit f59b925
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/google_symptoms/tests/conftest.py b/google_symptoms/tests/conftest.py
@@ -63,7 +63,9 @@
 county_data_gap = pd.read_csv(
     patch_input["county"], parse_dates=["date"])[keep_cols]
 
-covidcast_metadata = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata_backfill.csv",
+covidcast_backfill_metadata = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata_backfill.csv",
+                                          parse_dates=["max_time", "min_time", "max_issue", "last_update"])
+covidcast_metadata = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata.csv",
                                  parse_dates=["max_time", "min_time", "max_issue", "last_update"])
 
 NEW_DATE = "2024-02-20"
@@ -113,7 +115,8 @@ def run_as_module(params):
         makedirs("receiving")
 
     with mock.patch("delphi_google_symptoms.pull.initialize_credentials",
-                    return_value=None) as mock_credentials:
-        with mock.patch("pandas_gbq.read_gbq", side_effect=[
-                state_data, county_data]) as mock_read_gbq:
+                    return_value=None), \
+         mock.patch("pandas_gbq.read_gbq", side_effect=[state_data, county_data]), \
+         mock.patch("delphi_google_symptoms.pull.initialize_credentials", return_value=None), \
+         mock.patch("delphi_google_symptoms.date_utils.covidcast.metadata", return_value=covidcast_metadata):
             delphi_google_symptoms.run.run_module(params)