Exemple #1
0
import os
from copy import deepcopy
from os.path import join, exists
from tempfile import TemporaryDirectory

# third party
from delphi_utils import read_params
import numpy as np
import pandas as pd
import pytest

# first party
from delphi_claims_hosp.config import Config, GeoConstants
from delphi_claims_hosp.update_indicator import ClaimsHospIndicatorUpdater

CONFIG = Config()
CONSTANTS = GeoConstants()
PARAMS = read_params()
DATA_FILEPATH = PARAMS["input_file"]
DROP_DATE = pd.to_datetime(PARAMS["drop_date"])
OUTPATH = "test_data/"


class TestClaimsHospIndicatorUpdater:
    geo = "hrr"
    parallel = False
    weekday = False
    write_se = False
    prefix = "foo"
    small_test_data = pd.DataFrame({
        "num": [0, 100, 200, 300, 400, 500, 600, 100, 200, 300, 400, 500, 600],
    def test_values(self):
        conf = Config()

        assert conf.MIN_DEN == 100
        assert conf.MAX_BACKWARDS_PAD_LENGTH == 7