Remove app imports from test_app
@@ -2,11 +2,7 @@ import pytest
import pandas as pd
-from app import (current_hosp, doubling_time, relative_contact_rate,
- S, initial_infections,
- detection_prob, hospitalization_rates, I, R, beta, gamma, n_days, beta_decay,
- projection_admits, alt)
-from penn_chime.settings import DEFAULTS
+from app import (projection_admits, alt)
from penn_chime.models import sir, sim_sir, sim_sir_df
from penn_chime.presentation import display_header, new_admissions_chart
@@ -161,13 +157,8 @@ def test_sim_sir_df():
# assert initial_infections == known_infections
-def test_derived_variables():
- assert hospitalization_rates == (DEFAULTS.hosp.rate, DEFAULTS.icu.rate, DEFAULTS.vent.rate)
- # assert detection_prob == initial_infections / (current_hosp / DEFAULTS.market_share / DEFAULTS.hosp.rate)
-
-
def test_new_admissions_chart():
- chart = new_admissions_chart(alt, projection_admits, n_days - 10)
+ chart = new_admissions_chart(alt, projection_admits, 60 - 10)
assert type(chart) == alt.Chart
assert chart.data.iloc[1].Hospitalized < 1
# assert round(chart.data.iloc[49].ICU, 0) == 43
