1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
__author__ = 'fccoelho'
import unittest
import pytest
from matplotlib import pyplot as P
# import pyximport; pyximport.install(pyimport=True)
from epimodels.continuous.models import *
def test_SIR():
model = SIR()
model([1000, 1, 0], [0, 500], 1001, {'beta': .2, 'gamma': .1})
assert len(model.traces) == 4
# assert len(model.traces['time']) == 50
model.plot_traces()
P.show()
def test_SIR_with_t_eval():
model = SIR()
model([1000, 1, 0], [0, 500], 1001, {'beta': .2, 'gamma': .1}, t_eval=range(0, 500))
assert len(model.traces['S']) == 500
# assert len(model.traces['time']) == 50
def test_SIR1D():
model = SIR1D()
model([0], [0, 500], 100, {'R0': 1.5, 'gamma': .1, 'S0': 98})
# assert len(model.traces['R']) == 500
assert len(model.traces) == 2
model.plot_traces()
P.show()
def test_SIS():
model = SIS()
model([1000, 1], [0, 50], 1001, {'beta': 2, 'gamma': .1})
assert len(model.traces) == 3
# assert len(model.traces['time']) == 50
model.plot_traces()
P.show()
def test_SIRS():
model = SIRS()
model([1000, 1, 0], [0, 50], 1001, {'beta': 5, 'gamma': 1.9, 'xi': 0.05})
assert len(model.traces) == 4
# assert len(model.traces['time']) == 50
model.plot_traces()
P.show()
def test_SEIR():
model = SEIR()
model([1000, 0, 1, 0], [0, 50], 1001, {'beta': 5, 'gamma': 1.9, 'epsilon': 0.1})
# print(model.traces)
assert len(model.traces) == 5 # state variables plus time
# assert len(model.traces['time']) == 50
model.plot_traces()
P.show()
def test_SEQIAHR():
model = SEQIAHR()
model([.99, 0, 1e-6, 0, 0, 0, 0, 0], [0, 300], 1, {'chi': .7, 'phi': .01, 'beta': .5,
'rho': .05, 'delta': .1, 'gamma': .1,
'alpha': .33, 'mu': .03,
'p': .75, 'q': 50, 'r': 40
})
# print(model.traces)
assert len(model.traces) == 9 # state variables plus time
# assert len(model.traces['time']) == 50
model.plot_traces()
P.show()
# def test_SIS_with_cache():
# model = SIS()
# model([1000, 1], [0, 50], 1001, {'beta': 2, 'gamma': .1})
# tr1 = model.traces
# model([1000, 1], [0, 50], 1001, {'beta': 2, 'gamma': .1})
# tr2 = model.traces
# assert (tr1 == tr2)
|
