fix tests

Author: aysim319

google_symptoms/tests/conftest.py

@@ -41,7 +41,6 @@
 # from `bigquery-public-data.covid19_symptom_search.counties_daily_2020` # Counties by day; includes state and county name, + FIPS code
 # where timestamp(date) between timestamp("2020-07-15") and timestamp("2020-08-22")
 
-
 good_input = {
     "state": f"{TEST_DIR}/test_data/small_states_2020_07_15_2020_08_22.csv",
     "county": f"{TEST_DIR}/test_data/small_counties_2020_07_15_2020_08_22.csv"
@@ -59,6 +58,17 @@
 county_data = pd.read_csv(
     good_input["county"], parse_dates=["date"])[keep_cols]
 
+state_data_gap = pd.read_csv(patch_input["state"], parse_dates=["date"])[keep_cols]
+
+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"
+@pytest.fixture(scope="session")
+def logger():
+    return logging.getLogger()
 
 @pytest.fixture(scope="session")
 def params():
@@ -115,7 +125,7 @@ def run_as_module(params):
                     return_value=None), \
          mock.patch("pandas_gbq.read_gbq") as mock_read_gbq, \
          mock.patch("delphi_google_symptoms.pull.initialize_credentials", return_value=None), \
-         mock.patch("delphi_google_symptoms.run.Epidata.covidcast_meta", return_value=None) as mock_covidcast_meta:
+         mock.patch("delphi_google_symptoms.date_utils.Epidata.covidcast_meta", return_value=None) as mock_covidcast_meta:
         def side_effect(*args, **kwargs):
             if "symptom_search_sub_region_1_daily" in args[0]:
                 df = state_data

google_symptoms/tests/test_date_utils.py

@@ -4,13 +4,12 @@
 from freezegun import freeze_time
 from conftest import TEST_DIR, NEW_DATE
 
-import covidcast
+from delphi_epidata import Epidata
 
 from delphi_utils.validator.utils import lag_converter
 from delphi_google_symptoms.constants import FULL_BKFILL_START_DATE
 from delphi_google_symptoms.date_utils import generate_query_dates, generate_num_export_days, generate_patch_dates
 
-
 class TestDateUtils:
 
     @freeze_time("2021-01-05")
@@ -41,15 +40,15 @@ def test_generate_query_dates_custom(self):
 
     def test_generate_export_dates(self, params, logger, monkeypatch):
         metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata.csv")
-        monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
+        monkeypatch.setattr(Epidata, "covidcast_meta", lambda: metadata_df)
 
         num_export_days = generate_num_export_days(params, logger)
         expected_num_export_days = params["indicator"]["num_export_days"]
         assert num_export_days == expected_num_export_days
 
     def test_generate_export_dates_normal(self, params_w_no_date, logger, monkeypatch):
         metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata.csv")
-        monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
+        monkeypatch.setattr(Epidata, "covidcast_meta", lambda: metadata_df)
 
         num_export_days = generate_num_export_days(params_w_no_date, logger)
 
@@ -61,7 +60,7 @@ def test_generate_export_dates_normal(self, params_w_no_date, logger, monkeypatc
 
     def test_generate_export_date_missing(self, params_w_no_date, logger, monkeypatch):
         metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata_missing.csv")
-        monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
+        monkeypatch.setattr(Epidata, "covidcast_meta", lambda: metadata_df)
 
         num_export_days = generate_num_export_days(params_w_no_date, logger)
         expected_num_export_days = (date.today() - FULL_BKFILL_START_DATE.date()).days + 1

google_symptoms/tests/test_patch.py

@@ -53,7 +53,7 @@ def mocked_patch(self, params_):
         with mock_patch("delphi_google_symptoms.patch.read_params", return_value=params_), \
              mock_patch("delphi_google_symptoms.pull.pandas_gbq.read_gbq") as mock_read_gbq, \
              mock_patch("delphi_google_symptoms.pull.initialize_credentials", return_value=None), \
-             mock_patch("delphi_google_symptoms.date_utils.covidcast.metadata", return_value=covidcast_metadata), \
+             mock_patch("delphi_google_symptoms.date_utils.Epidata.covidcast_meta", return_value=covidcast_metadata), \
              mock_patch("delphi_google_symptoms.run.GEO_RESOLUTIONS", new=["state"]):
             def side_effect(*args, **kwargs):
                 if "symptom_search_sub_region_1_daily" in args[0]: