|
4 | 4 | from freezegun import freeze_time
|
5 | 5 | from conftest import TEST_DIR, NEW_DATE
|
6 | 6 |
|
| 7 | +import covidcast |
| 8 | + |
7 | 9 | from delphi_utils.validator.utils import lag_converter
|
8 | 10 | from delphi_google_symptoms.constants import FULL_BKFILL_START_DATE
|
9 | 11 | from delphi_google_symptoms.date_utils import generate_query_dates, generate_num_export_days, generate_patch_dates
|
| 12 | + |
| 13 | + |
10 | 14 | class TestDateUtils:
|
11 | 15 |
|
12 | 16 | @freeze_time("2021-01-05")
|
@@ -36,31 +40,29 @@ def test_generate_query_dates_custom(self):
|
36 | 40 | assert set(output) == set(expected)
|
37 | 41 |
|
38 | 42 | def test_generate_export_dates(self, params, logger, monkeypatch):
|
39 |
| - import covidcast |
40 | 43 | metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata.csv")
|
41 | 44 | monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
|
42 |
| - num_export_days = generate_num_export_days(params, logger) |
43 | 45 |
|
| 46 | + num_export_days = generate_num_export_days(params, logger) |
44 | 47 | expected_num_export_days = params["indicator"]["num_export_days"]
|
45 |
| - |
46 | 48 | assert num_export_days == expected_num_export_days
|
47 | 49 |
|
48 | 50 | def test_generate_export_dates_normal(self, params_w_no_date, logger, monkeypatch):
|
49 |
| - import covidcast |
50 | 51 | metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata.csv")
|
51 | 52 | monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
|
| 53 | + |
52 | 54 | num_export_days = generate_num_export_days(params_w_no_date, logger)
|
53 | 55 |
|
54 |
| - max_expected_lag = lag_converter(params_w_no_date["validation"]["common"].get("max_expected_lag", {"all": 4})) |
| 56 | + max_expected_lag = lag_converter(params_w_no_date["validation"]["common"]["max_expected_lag"]) |
55 | 57 | global_max_expected_lag = max(list(max_expected_lag.values()))
|
56 |
| - expected_num_export_days = params_w_no_date["validation"]["common"].get("span_length", 14) + global_max_expected_lag |
| 58 | + expected_num_export_days = params_w_no_date["validation"]["common"]["span_length"] + global_max_expected_lag |
57 | 59 |
|
58 | 60 | assert num_export_days == expected_num_export_days
|
59 | 61 |
|
60 | 62 | def test_generate_export_date_missing(self, params_w_no_date, logger, monkeypatch):
|
61 |
| - import covidcast |
62 | 63 | metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata_missing.csv")
|
63 | 64 | monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df)
|
| 65 | + |
64 | 66 | num_export_days = generate_num_export_days(params_w_no_date, logger)
|
65 | 67 | expected_num_export_days = (date.today() - FULL_BKFILL_START_DATE.date()).days + 1
|
66 | 68 | assert num_export_days == expected_num_export_days
|
@@ -89,10 +91,7 @@ def generate_expected_start_end_dates(self, params_, issue_date):
|
89 | 91 | "export_end_date": max(dates[6:21])
|
90 | 92 | }
|
91 | 93 |
|
92 |
| - def test_generate_patch_dates(self, params_w_patch, logger, monkeypatch): |
93 |
| - import covidcast |
94 |
| - metadata_df = pd.read_csv(f"{TEST_DIR}/test_data/covid_metadata_missing.csv") |
95 |
| - monkeypatch.setattr(covidcast, "metadata", lambda: metadata_df) |
| 94 | + def test_generate_patch_dates(self, params_w_patch, logger): |
96 | 95 | max_expected_lag = lag_converter(params_w_patch["validation"]["common"]["max_expected_lag"])
|
97 | 96 | global_max_expected_lag = max(list(max_expected_lag.values()))
|
98 | 97 | num_export_days = params_w_patch["validation"]["common"]["span_length"]
|
|
0 commit comments