def test_main():
"""Run main function below."""
# Get US surge data
us_surge = Surge()
# Set parameters
us_surge.end_date = '5/15/20' # set end date wanted
us_surge.ignore_last_n_days = 0 # allow for data repo to be updated
us_surge.min_n_cases_abs = 500 # min # of absolute cases for analysis
us_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic Lower
# Respiratory Diseases per year: 41(2019)
print('')
print('# of states/distric: ', len(us_surge.names))
print('# of days: ', us_surge.dates.shape[0])
# Fit data to all states
fit_data = us_surge.multi_fit_data(verbose=True,
plot=True,
save_plots=True)
# Plot all data in one plot
us_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
# Plot all fit data in one plot
us_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = us_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s' % (k, bins[k]))
# Use bins to create groups of states based on surge period
state_groups = dict()
for (sort_key, data) in fit_data:
state = data[0]
#param_vec = data[3]
key = us_surge.get_bin_id(sort_key, bins)
if key in state_groups:
state_groups[key].append(state)
else:
state_groups[key] = list()
state_groups[key].append(state)
state_groups = [
state_groups[k] for k in sorted(state_groups.keys(), reverse=False)
]
print('')
print('*****************************************************************')
print(' Country Groups ')
print('*****************************************************************')
for grp in state_groups:
print(' Group %i %s' % (state_groups.index(grp), grp))
print('')
print('Testing results:')
try:
assert_equal(len(state_groups), 7)
assert_equal(state_groups[0], ['New York', 'Virginia'])
assert_equal(state_groups[1],
['Massachusetts', 'Connecticut', 'Michigan'])
assert_equal(state_groups[2], [
'New Jersey', 'Pennsylvania', 'Louisiana', 'Minnesota', 'Maryland'
])
assert_equal(state_groups[3], ['North Carolina'])
assert_equal(
state_groups[4],
['Indiana', 'Florida', 'Georgia', 'Wisconsin', 'Missouri'])
assert_equal(
state_groups[5],
['Colorado', 'Ohio', 'Illinois', 'California', 'Washington'])
assert_equal(state_groups[6], ['Alabama', 'Mississippi'])
except AssertionError as err:
print('Warning: ', err)
else:
print('all tests passed')
# Plot the normalized surge for groups of states
us_surge.plot_group_fit_data(state_groups, fit_data, save=True)
# Plot the surge period for all grouped states
us_surge.plot_group_surge_periods(fit_data, bins, save=True)
def test_main():
"""Run main function below."""
# Get US surge data
sub_locale = 'North Carolina'
c_surge = Surge(locale='US', sub_locale=sub_locale)
print('')
print('State : ', sub_locale)
print('# of counties: ', len(c_surge.names))
# Set parameters
c_surge.end_date = '5/15/20' # set end date wanted
c_surge.ignore_last_n_days = 0 # allow for data repo to be updated
c_surge.min_n_cases_abs = 25 # min # of absolute cases for analysis
c_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic Lower
# Respiratory Diseases per year: 41 (2019)
# Fit data to all counties/cities
fit_data = c_surge.multi_fit_data(verbose=True, plot=True, save_plots=True)
print('# of fittings done = ', len(fit_data))
# Plot all data in one plot
c_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
if not fit_data:
print('Done here...')
return
# Plot all fit data in one plot
c_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = c_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s'%(k, bins[k]))
# Use bins to create groups of counties/cities based on surge period
county_groups = dict()
for (sort_key, data) in fit_data:
county = data[0]
#param_vec = data[3]
key = c_surge.get_bin_id(sort_key, bins)
if key in county_groups:
county_groups[key].append(county)
else:
county_groups[key] = list()
county_groups[key].append(county)
county_groups = [county_groups[k] for k in
sorted(county_groups.keys(), reverse=False)]
print('')
print('*****************************************************************')
print(' County Groups ')
print('*****************************************************************')
for grp in county_groups:
print(' Group %i %s'%(county_groups.index(grp), grp))
print('')
print('Testing results:')
try:
assert_equal(len(county_groups), 4)
assert_equal(county_groups[0], ['Rowan'])
assert_equal(county_groups[1], ['Orange', 'Durham'])
assert_equal(county_groups[2], ['Mecklenburg'])
assert_equal(county_groups[3], ['Guilford'])
except AssertionError as err:
print('Warning: ', err)
else:
print('all tests passed')
# Plot the normalized surge for groups of counties
c_surge.plot_group_fit_data(county_groups, fit_data, save=True)
# Plot the surge period for all grouped counties
c_surge.plot_group_surge_periods(fit_data, bins, save=True)
def main():
"""Main function executed at the bottom."""
# Get US surge data
us_surge = Surge()
# Set parameters
us_surge.end_date = '4/20/20' # set end date wanted
us_surge.end_date = None # get all the data available
us_surge.ignore_last_n_days = 2 # allow for data repo to be updated
us_surge.min_n_cases_abs = 500 # min # of absolute cases for analysis
us_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic
# Lower Respiratory Diseases per year:
# 41 (2019)
print('')
print('# of states/distric: ', len(us_surge.names))
print('# of days: ', us_surge.dates.shape[0])
# Fit data to all states
fit_data = us_surge.multi_fit_data(verbose=True,
plot=True,
save_plots=True)
# Plot all data in one plot
us_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
# Plot all fit data in one plot
us_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = us_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s' % (k, bins[k]))
# Use bins to create groups of states based on surge period
state_groups = dict()
for (sort_key, data) in fit_data:
state = data[0]
#param_vec = data[3]
key = us_surge.get_bin_id(sort_key, bins)
if key in state_groups:
state_groups[key].append(state)
else:
state_groups[key] = list()
state_groups[key].append(state)
state_groups = [
state_groups[k] for k in sorted(state_groups.keys(), reverse=False)
]
print('')
print('*****************************************************************')
print(' Country Groups ')
print('*****************************************************************')
for grp in state_groups:
print(' Group %i %s' % (state_groups.index(grp), grp))
# Plot the normalized surge for groups of states
us_surge.plot_group_fit_data(state_groups, fit_data, save=True)
# Plot the surge period for all grouped states
us_surge.plot_group_surge_periods(fit_data, bins, save=True)
def main():
"""Main function executed at the bottom."""
# Get US surge data
sub_locale = 'North Carolina'
c_surge = Surge(locale='US', sub_locale=sub_locale)
print('')
print('State : ', sub_locale)
print('# of counties: ', len(c_surge.names))
# Set parameters
c_surge.end_date = '4/20/20' # set end date wanted
c_surge.end_date = None # get all the data available
c_surge.ignore_last_n_days = 2 # allow for data repo to be updated
c_surge.min_n_cases_abs = 25 # min # of absolute cases for analysis
c_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic Lower
# Respiratory Diseases per year: 41 (2019)
# Fit data to all counties/cities
fit_data = c_surge.multi_fit_data(verbose=True, plot=True, save_plots=True)
print('# of fittings done = ', len(fit_data))
# Plot all data in one plot
c_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
if not fit_data:
print('Done here...')
return
# Plot all fit data in one plot
c_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = c_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s'%(k, bins[k]))
# Use bins to create groups of counties/cities based on surge period
county_groups = dict()
for (sort_key, data) in fit_data:
county = data[0]
#param_vec = data[3]
key = c_surge.get_bin_id(sort_key, bins)
if key in county_groups:
county_groups[key].append(county)
else:
county_groups[key] = list()
county_groups[key].append(county)
county_groups = [county_groups[k] for k in
sorted(county_groups.keys(), reverse=False)]
print('')
print('*****************************************************************')
print(' County Groups ')
print('*****************************************************************')
for grp in county_groups:
print(' Group %i %s'%(county_groups.index(grp), grp))
# Plot the normalized surge for groups of counties
c_surge.plot_group_fit_data(county_groups, fit_data, save=True)
# Plot the surge period for all grouped counties
c_surge.plot_group_surge_periods(fit_data, bins, save=True)
def test_main():
"""Main function executed at the bottom."""
# Get US surge data
us_surge = Surge()
#states = [ a for (a,b) in
# sorted( zip(us_surge.names, us_surge.cases[-1,:]),
# key = lambda entry: entry[1], reverse=True )]
# Set parameters
us_surge.end_date = '4/20/20' # set end date wanted
us_surge.end_date = None # get all the data available
us_surge.ignore_last_n_days = 2 # allow for data repo to be updated
us_surge.min_n_cases_abs = 500 # min # of absolute cases for analysis
us_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic
# Lower Respiratory Diseases per
# year: 41 (2019)
print('')
print('# of states/distric: ', len(us_surge.names))
print('# of days: ', us_surge.dates.shape[0])
# Fit data to all states of fully-evolved surge
fit_data = us_surge.multi_fit_data() # silent
states = list()
print('')
for (i, (sort_key, data)) in enumerate(fit_data):
name = data[0]
states.append(name)
print('%2i) %15s: surge period %1.2f [day]'%(i, name, sort_key))
surge_periods = list() # collect surge period of all counties/towns
total_n_counties = 0 # count all counties/towns inspected w/ nonzero cases
for state in states:
print('')
print('**************************************************************')
print(' ', state)
print('**************************************************************')
c_surge = Surge(locale='US', sub_locale=state)
print('# of counties: ', len(c_surge.names))
(ids,) = np.where(c_surge.cases[-1, :] > 0)
total_n_counties += ids.size
# Set parameters
c_surge.end_date = '4/20/20' # set end date wanted
c_surge.end_date = None # get all the data available
c_surge.ignore_last_n_days = 2 # allow for data repo to be updated
c_surge.min_n_cases_abs = 100 # min # of absolute cases for analysis
c_surge.deaths_100k_minimum = 41 # US death per 100,000 for Chronic
# Lower Respiratory Diseases per
# year: 41 (2019)
# Fit data to all counties/cities
fit_data = c_surge.multi_fit_data(verbose=False, plot=True,
save_plots=True)
print('# of fittings done = ', len(fit_data))
if not fit_data:
print('Done here...')
continue
print('')
for (sort_key, data) in fit_data:
name = data[0]
surge_periods.append(sort_key)
print('%15s: surge period %1.2f [day]'%(name, sort_key))
# Create clustering bins based on surge period
bins = c_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('--------------------------------------------------------------')
print(' Bins ')
print('--------------------------------------------------------------')
for k in sorted(bins.keys()):
print(' Bin %i %s'%(k, bins[k]))
# Use bins to create groups of counties/cities based on surge period
county_groups = dict()
for (sort_key, data) in fit_data:
county = data[0]
#param_vec = data[3]
key = c_surge.get_bin_id(sort_key, bins)
if key in county_groups:
county_groups[key].append(county)
else:
county_groups[key] = list()
county_groups[key].append(county)
county_groups = [county_groups[k] for k in
sorted(county_groups.keys(), reverse=False)]
print('')
print('--------------------------------------------------------------')
print(' County Groups ')
print('--------------------------------------------------------------')
for grp in county_groups:
print(' Group %i %s'%(county_groups.index(grp), grp))
# Plot the surge period for all grouped counties
c_surge.plot_group_surge_periods(fit_data, bins, save=True)
print('')
print('')
print('')
print('')
print('Total # of counties/towns with surge period = ', len(surge_periods))
print('Average surge period %1.2f [day], std %1.2f'%
(np.mean(np.array(surge_periods)), np.std(np.array(surge_periods))))
print('Total # of inspected counties/towns w/ non-zero cases = %4i'%
total_n_counties)
gold_n_counties = 99
gold_ave_surge_period = 25
gold_total_n_counties = 1158
try:
assert_almost_equal(len(surge_periods), gold_n_counties, delta=3)
assert_almost_equal(np.mean(np.array(surge_periods)),
gold_ave_surge_period, delta=5)
assert_almost_equal(total_n_counties, gold_total_n_counties, delta=10)
except AssertionError as err:
print('Warning: ', err)
else:
print('all tests passed')
print('')
def main():
"""Main function executed at the bottom."""
# Get US surge data
g_surge = Surge('global')
# Set parameters
g_surge.end_date = '4/20/20' # set end date wanted
g_surge.end_date = None # get all the data available
g_surge.ignore_last_n_days = 2 # allow for data repo to be updated
g_surge.min_n_cases_abs = 2500 # min # of absolute cases for analysis
print('# of countries: ', g_surge.cases.shape[1])
print('# of days: ', g_surge.cases.shape[0])
# Fit data to all states
fit_data = g_surge.multi_fit_data(blocked_list=['China'],
verbose=True,
plot=True,
save_plots=True)
# Plot all data in one plot
g_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
# Plot all fit data in one plot
g_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = g_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s' % (k, bins[k]))
# Use bins to create groups of countries based on surge period
country_groups = dict()
for (sort_key, data) in fit_data:
country = data[0]
key = g_surge.get_bin_id(sort_key, bins)
if key in country_groups:
country_groups[key].append(country)
else:
country_groups[key] = list()
country_groups[key].append(country)
country_groups = [
country_groups[k] for k in sorted(country_groups.keys(), reverse=False)
]
print('')
print('*****************************************************************')
print(' Country Groups ')
print('*****************************************************************')
for gri in country_groups:
print(' Group %i %s' % (country_groups.index(gri), gri))
# Plot the normalized surge for groups of countries
g_surge.plot_group_fit_data(country_groups, fit_data, save=True)
# Plot the surge period for all grouped states
g_surge.plot_group_surge_periods(fit_data, bins, save=True)
def test_main():
"""Run main function below."""
# Get US surge data
g_surge = Surge('global')
# Set parameters
g_surge.end_date = '5/15/20' # set end date wanted
g_surge.ignore_last_n_days = 0 # allow for data repo to be corrected
g_surge.min_n_cases_abs = 2500 # min # of absolute cases for analysis
print('# of countries: ', g_surge.cases.shape[1])
print('# of days: ', g_surge.cases.shape[0])
# Fit data to all states
fit_data = g_surge.multi_fit_data(blocked_list=['China'],
verbose=True, plot=True, save_plots=True)
# Plot all data in one plot
g_surge.plot_multi_fit_data(fit_data, 'experimental', save=True)
# Plot all fit data in one plot
g_surge.plot_multi_fit_data(fit_data, 'fit', save=True)
# Create clustering bins based on surge period
bins = g_surge.fit_data_bins(fit_data, 2, 'surge_period')
print('')
print('*****************************************************************')
print(' Bins ')
print('*****************************************************************')
for k in sorted(bins.keys()):
print(' Bin %i %s'%(k, bins[k]))
# Use bins to create groups of countries based on surge period
country_groups = dict()
for (sort_key, data) in fit_data:
country = data[0]
#param_vec = data[3]
key = g_surge.get_bin_id(sort_key, bins)
if key in country_groups:
country_groups[key].append(country)
else:
country_groups[key] = list()
country_groups[key].append(country)
country_groups = [country_groups[k] for k in
sorted(country_groups.keys(), reverse=False)]
print('')
print('*****************************************************************')
print(' Country Groups ')
print('*****************************************************************')
for grp in country_groups:
print(' Group %i %s'%(country_groups.index(grp), grp))
print('')
print('Testing results:')
try:
assert_equal(len(country_groups), 6)
assert_equal(country_groups[0], ['Belgium', 'France'])
assert_equal(country_groups[1], ['Germany', 'Turkey'])
assert_equal(country_groups[2], ['Spain', 'Netherlands',
'United Kingdom', 'Canada'])
assert_equal(country_groups[3], ['Ecuador'])
assert_equal(country_groups[4], ['Sweden', 'US', 'Italy'])
assert_equal(country_groups[5], ['Iran'])
except AssertionError as err:
print('Warning: ', err)
else:
print('all tests passed')
# Plot the normalized surge for groups of countries
g_surge.plot_group_fit_data(country_groups, fit_data, save=True)
# Plot the surge period for all grouped states
g_surge.plot_group_surge_periods(fit_data, bins, save=True)