5
5
import numpy as np
6
6
import altair as alt
7
7
8
- from penn_chime .models import sir , sim_sir , sim_sir_df
9
- from penn_chime .parameters import Parameters
10
- from penn_chime .presentation import display_header , new_admissions_chart , admitted_patients_chart
11
- from penn_chime .settings import DEFAULTS
12
- from penn_chime .defaults import RateLos
8
+ from src . penn_chime .models import sir , sim_sir , sim_sir_df
9
+ from src . penn_chime .parameters import Parameters
10
+ from src . penn_chime .presentation import display_header , new_admissions_chart , admitted_patients_chart
11
+ from src . penn_chime .settings import DEFAULTS
12
+ from src . penn_chime .defaults import RateLos
13
13
14
14
15
15
# set up
@@ -162,7 +162,7 @@ def test_sim_sir_df():
162
162
163
163
164
164
def test_new_admissions_chart ():
165
- projection_admits = pd .read_csv ('src/ tests/projection_admits.csv' )
165
+ projection_admits = pd .read_csv ('tests/projection_admits.csv' )
166
166
chart = new_admissions_chart (alt , projection_admits , 60 - 10 )
167
167
assert isinstance (chart , alt .Chart )
168
168
assert chart .data .iloc [1 ].Hospitalized < 1
@@ -177,7 +177,7 @@ def test_new_admissions_chart():
177
177
178
178
179
179
def test_admitted_patients_chart ():
180
- census_df = pd .read_csv ('src/ tests/census_df.csv' )
180
+ census_df = pd .read_csv ('tests/census_df.csv' )
181
181
chart = admitted_patients_chart (alt , census_df , 60 - 10 )
182
182
assert isinstance (chart , alt .Chart )
183
183
assert chart .data .iloc [1 ]['Hospital Census' ] == 1
0 commit comments