|
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