63
63
county_data_gap = pd .read_csv (
64
64
patch_input ["county" ], parse_dates = ["date" ])[keep_cols ]
65
65
66
- covidcast_metadata = pd .read_csv (f"{ TEST_DIR } /test_data/covid_metadata_backfill.csv" ,
66
+ covidcast_backfill_metadata = pd .read_csv (f"{ TEST_DIR } /test_data/covid_metadata_backfill.csv" ,
67
+ parse_dates = ["max_time" , "min_time" , "max_issue" , "last_update" ])
68
+ covidcast_metadata = pd .read_csv (f"{ TEST_DIR } /test_data/covid_metadata.csv" ,
67
69
parse_dates = ["max_time" , "min_time" , "max_issue" , "last_update" ])
68
70
69
71
NEW_DATE = "2024-02-20"
@@ -113,7 +115,8 @@ def run_as_module(params):
113
115
makedirs ("receiving" )
114
116
115
117
with mock .patch ("delphi_google_symptoms.pull.initialize_credentials" ,
116
- return_value = None ) as mock_credentials :
117
- with mock .patch ("pandas_gbq.read_gbq" , side_effect = [
118
- state_data , county_data ]) as mock_read_gbq :
118
+ return_value = None ), \
119
+ mock .patch ("pandas_gbq.read_gbq" , side_effect = [state_data , county_data ]), \
120
+ mock .patch ("delphi_google_symptoms.pull.initialize_credentials" , return_value = None ), \
121
+ mock .patch ("delphi_google_symptoms.date_utils.covidcast.metadata" , return_value = covidcast_metadata ):
119
122
delphi_google_symptoms .run .run_module (params )
0 commit comments