Coverage for models / Covid / ensemble_training / kinds.py: 100%

1from enum import IntEnum 

2 

3# Agent kinds and their order used in the SEIRD+ model 

4 

5 

6class Compartments(IntEnum): 

7 susceptible = 0 

8 exposed = 1 

9 infected = 2 

10 recovered = 3 

11 symptomatic = 4 

12 hospitalized = 5 

13 critical = 6 

14 deceased = 7 

15 quarantine_S = 8 

16 quarantine_E = 9 

17 quarantine_I = 10 

18 contact_traced = 11