def examplew2():
pars = {'use_waning': True, 'n_days': 180}
sim = cv.Sim(pars=pars)
scenarios = {
'base': {
'name': 'baseline',
'pars': {}
},
'scen1': {
'name': '1 wave',
'pars': {
'interventions':
[cv.historical_wave(prob=0.25, days_prior=50)]
}
},
'scen2': {
'name': '2 waves',
'pars': {
'interventions':
[cv.historical_wave(prob=[0.20, 0.05], days_prior=[360, 50])]
}
}
}
metapars = cv.make_metapars()
metapars.update({'n_runs': 8})
scens = cv.Scenarios(sim=sim, scenarios=scenarios, metapars=metapars)
scens.run()
scens.plot()
def test_historical():
pfizer = cv.prior_immunity(vaccine='pfizer', days=np.arange(-30, 0), prob=0.007)
wave = cv.historical_wave(120, 0.05)
sim1 = cv.Sim(base_pars, interventions=pfizer).run()
sim2 = cv.Sim(base_pars, interventions=wave).run()
with pytest.raises(RuntimeError):
cv.Sim(base_pars, pop_scale=5, interventions=wave).run()
with pytest.raises(ValueError):
cv.Sim(base_pars, interventions=cv.historical_wave(120, 0.05, variant='invalid')).run()
return sim1, sim2
def examplew3():
pars = {'use_waning': True, 'n_days': 1}
sim = cv.Sim(pars=pars)
sim['interventions'] += [
cv.historical_wave(prob=[0.05, 0.05], days_prior=[100, 50])
]
sim['analyzers'] += [cv.nab_histogram(days=[0])]
sim.run()
sim.plot('variants')
sim['analyzers'][0].plot()
def examplew1():
# run single sim
pars = {'use_waning': True, 'rand_seed': 1}
variants = [cv.variant('delta', days=15, n_imports=10)]
sim = cv.Sim(pars=pars, variants=variants)
sim['interventions'] += [
cv.historical_wave(variant='wild',
prob=[0.05, 0.05],
days_prior=[150, 50])
]
sim.run()
sim.plot()
sim.plot('variants')
def examplew4():
with pytest.raises(ValueError):
cv.Sim(use_waning=True,
interventions=[cv.historical_wave(120, 0.05, variant='delta')
]).run().plot()
def examplew0():
cv.Sim(use_waning=True,
interventions=[cv.historical_wave(120, 0.05)]).run().plot()