# standard
import numpy as np
import numpy.random as nr
import pandas as pd
# first party
from delphi_changehc.config import Config
from delphi_changehc.load_data import load_combined_data
from delphi_changehc.sensor import CHCSensor
# third party
from delphi_utils import read_params
CONFIG = Config()
PARAMS = read_params()
COVID_FILEPATH = PARAMS["input_covid_file"]
DENOM_FILEPATH = PARAMS["input_denom_file"]
DROP_DATE = pd.to_datetime(PARAMS["drop_date"])
class TestLoadData:
combined_data = load_combined_data(DENOM_FILEPATH, COVID_FILEPATH,
DROP_DATE, "fips")
def test_backfill(self):
num0 = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8],
dtype=float).reshape(-1, 1)
den0 = np.array([0, 10, 10, 10, 10, 10, 10, 100, 101], dtype=float)
num1, den1 = CHCSensor.backfill(num0, den0, k=7, min_visits_to_fill=0)
assert np.array_equal(num0, num1)
assert np.array_equal(den0, den1)
def test_values(self):
conf = Config()
assert conf.MIN_DEN == 100
assert conf.MAX_BACKFILL_WINDOW == 7