def test_get_data():
beds_data = CovidCareMapBeds.local().beds()
data = beds_data.get_record_for_state('MA')
assert data
data = beds_data.get_record_for_state('NOTSTATE')
assert not data
def test_nyc_aggregation():
beds_data = CovidCareMapBeds.local().beds()
data = beds_data.get_record_for_fips(
custom_aggregations.NEW_YORK_COUNTY_FIPS)
# Check to make sure that beds occupancy rates are below 1,
# signaling that it is properly combining occupancy rates.
assert data["all_beds_occupancy_rate"] < 1
assert data["icu_occupancy_rate"] < 1
def test_get_data():
beds_data = CovidCareMapBeds.local().beds()
data = beds_data.get_record_for_fips("25")
assert data
data = beds_data.get_record_for_fips("NOTSTATE")
assert not data
data = beds_data.get_record_for_fips("99")
assert not data
def __init__(self,
fips,
N_samples,
t_list,
I_initial=1,
suppression_policy=None):
# Caching globally to avoid relatively significant performance overhead
# of loading for each county.
global beds_data, population_data
if not beds_data or not population_data:
beds_data = CovidCareMapBeds.local().beds()
population_data = FIPSPopulation.local().population()
self.fips = fips
self.agg_level = AggregationLevel.COUNTY if len(
self.fips) == 5 else AggregationLevel.STATE
self.N_samples = N_samples
self.I_initial = I_initial
self.suppression_policy = suppression_policy
self.t_list = t_list
if self.agg_level is AggregationLevel.COUNTY:
self.county_metadata = load_data.load_county_metadata().set_index(
'fips').loc[fips].to_dict()
self.state_abbr = us.states.lookup(
self.county_metadata['state']).abbr
self.population = population_data.get_record_for_fips(
fips=self.fips)[CommonFields.POPULATION]
# TODO: Some counties do not have hospitals. Likely need to go to HRR level..
self._beds_data = beds_data.get_record_for_fips(fips)
else:
self.state_abbr = us.states.lookup(fips).abbr
self.population = population_data.get_record_for_state(
self.state_abbr)[CommonFields.POPULATION]
self._beds_data = beds_data.get_record_for_state(self.state_abbr)
def test_pr_aggregation():
beds_data = CovidCareMapBeds.local().beds()
data = beds_data.get_record_for_fips("72")
assert data
assert data["all_beds_occupancy_rate"] < 1
assert data["icu_occupancy_rate"] < 1
def get_beds_data():
return CovidCareMapBeds.local().beds()