Skip to content

Commit 7e558a2

Browse files
committed
remove overzealous date mocking
1 parent cc67b7f commit 7e558a2

File tree

1 file changed

+5
-16
lines changed

1 file changed

+5
-16
lines changed

google_symptoms/tests/conftest.py

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -51,15 +51,6 @@
5151
good_input["county"], parse_dates=["date"])[keep_cols]
5252

5353

54-
# Set up fake list of dates to fetch.
55-
dates = [
56-
"20200726",
57-
"20200811"
58-
]
59-
60-
date_list = [datetime.strptime(date, "%Y%m%d") for date in dates]
61-
62-
6354
@pytest.fixture(scope="session")
6455
def run_as_module():
6556
params = {
@@ -81,10 +72,8 @@ def run_as_module():
8172
else:
8273
makedirs("receiving")
8374

84-
with mock.patch("delphi_google_symptoms.pull.get_date_range",
85-
return_value=date_list) as mock_all_dates:
86-
with mock.patch("delphi_google_symptoms.pull.initialize_credentials",
87-
return_value=None) as mock_credentials:
88-
with mock.patch("pandas_gbq.read_gbq", side_effect=[
89-
state_data, county_data]) as mock_read_gbq:
90-
delphi_google_symptoms.run.run_module(params)
75+
with mock.patch("delphi_google_symptoms.pull.initialize_credentials",
76+
return_value=None) as mock_credentials:
77+
with mock.patch("pandas_gbq.read_gbq", side_effect=[
78+
state_data, county_data]) as mock_read_gbq:
79+
delphi_google_symptoms.run.run_module(params)

0 commit comments

Comments
 (0)