From c58f127f801e7a54ff3ca449b862d8226883205f Mon Sep 17 00:00:00 2001
From: Gregory Wagner <wagner.greg@gmail.com>
Date: Wed, 27 May 2020 07:49:14 -0400
Subject: [PATCH 1/2] Adds basic regression test

---
 test/test_king_county_scenario.py | 58 +++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 test/test_king_county_scenario.py

diff --git a/test/test_king_county_scenario.py b/test/test_king_county_scenario.py
new file mode 100644
index 00000000..ba390215
--- /dev/null
+++ b/test/test_king_county_scenario.py
@@ -0,0 +1,58 @@
+# Get the epiforecast package onto our path:
+import os, sys; sys.path.append(os.path.join(".."))
+
+import numpy as np
+
+from epiforecast.scenarios import king_county_transition_rates
+
+def get_rates(rates_name):
+    return getattr(king_county_transition_rates(population=3, random_seed=1234), rates_name)
+
+transition_rates = king_county_transition_rates(population=3, random_seed=1234)
+ 
+for name in (
+              'latent_periods',
+              'community_infection_periods',
+              'hospital_infection_periods',
+              'hospitalization_fraction',
+              'community_mortality_fraction',
+              'hospital_mortality_fraction',
+            ):
+
+    print(get_rates(name))
+
+#def test_latent_periods(): 
+#    assert (
+#            get_rates('latent_periods') 
+#                == np.array([6.59337662, 6.63560047, 4.03092229])
+#           ).all()
+
+#def test_community_infection_periods(): 
+#    assert (
+#            get_rates('community_infection_periods') 
+#                == np.array([5.7768707, 1.91422953, 7.96380771])
+#           ).all()
+#
+#def test_hospital_infection_periods(): 
+#    assert (
+#            get_rates('hospital_infection_periods') 
+#                == np.array([3.60376111, 8.10037068, 6.84607501])
+#           ).all()
+#
+#def test_hospitalization_fraction(): 
+#    assert (
+#            get_rates('hospitalization_fraction') 
+#                == np.array([3.67966623e-06, 3.78385250e-01, 4.27390020e-02])
+#           ).all()
+#
+#def test_community_mortality_fraction(): 
+#    assert (
+#            get_rates('community_mortality_fraction') 
+#                == np.array([5.79223912e-90, 5.03197685e-38, 0.00000000e+00])
+#           ).all()
+#
+#def test_hospital_mortality_fraction(): 
+#    assert (
+#            get_rates('hospital_mortality_fraction') 
+#                == np.array([[4.38776678e-76, 7.30522015e-07, 5.15894221e-02]])
+#           ).all()

From 3f42e7f86cb24fe5dc1057d51fe9e7d04439b74a Mon Sep 17 00:00:00 2001
From: Gregory Wagner <wagner.greg@gmail.com>
Date: Wed, 27 May 2020 17:07:39 -0400
Subject: [PATCH 2/2] Fixes up king county scenario test

---
 test/test_king_county_scenario.py | 52 ++++++++++---------------------
 1 file changed, 17 insertions(+), 35 deletions(-)

diff --git a/test/test_king_county_scenario.py b/test/test_king_county_scenario.py
index ba390215..a69c78cc 100644
--- a/test/test_king_county_scenario.py
+++ b/test/test_king_county_scenario.py
@@ -21,38 +21,20 @@ def get_rates(rates_name):
 
     print(get_rates(name))
 
-#def test_latent_periods(): 
-#    assert (
-#            get_rates('latent_periods') 
-#                == np.array([6.59337662, 6.63560047, 4.03092229])
-#           ).all()
-
-#def test_community_infection_periods(): 
-#    assert (
-#            get_rates('community_infection_periods') 
-#                == np.array([5.7768707, 1.91422953, 7.96380771])
-#           ).all()
-#
-#def test_hospital_infection_periods(): 
-#    assert (
-#            get_rates('hospital_infection_periods') 
-#                == np.array([3.60376111, 8.10037068, 6.84607501])
-#           ).all()
-#
-#def test_hospitalization_fraction(): 
-#    assert (
-#            get_rates('hospitalization_fraction') 
-#                == np.array([3.67966623e-06, 3.78385250e-01, 4.27390020e-02])
-#           ).all()
-#
-#def test_community_mortality_fraction(): 
-#    assert (
-#            get_rates('community_mortality_fraction') 
-#                == np.array([5.79223912e-90, 5.03197685e-38, 0.00000000e+00])
-#           ).all()
-#
-#def test_hospital_mortality_fraction(): 
-#    assert (
-#            get_rates('hospital_mortality_fraction') 
-#                == np.array([[4.38776678e-76, 7.30522015e-07, 5.15894221e-02]])
-#           ).all()
+def test_latent_periods(): 
+    assert get_rates('latent_periods')[1] == 6.635600468016812 
+
+def test_community_infection_periods(): 
+    assert get_rates('community_infection_periods')[1] == 1.9142295330323456
+
+def test_hospital_infection_periods(): 
+    assert get_rates('hospital_infection_periods')[1] == 8.100370683584359 
+
+def test_hospitalization_fraction(): 
+    assert get_rates('hospitalization_fraction')[1] == 0.37838525036214415
+
+def test_community_mortality_fraction(): 
+    assert get_rates('community_mortality_fraction')[1] == 5.031976849805044e-38
+            
+def test_hospital_mortality_fraction(): 
+    assert get_rates('hospital_mortality_fraction')[1] == 7.305220148909824e-07