Fix missing model parameter

Author: jlubken

src/penn_chime/presentation.py

@@ -76,9 +76,9 @@ def display_header(st, m, p):
             r_naught=m.r_naught,
             doubling_time=p.doubling_time,
             relative_contact_rate=p.relative_contact_rate,
-            r_t=p.r_t,
-            doubling_time_t=abs(p.doubling_time_t),
-            impact_statement=("halves the infections every" if p.r_t < 1 else "reduces the doubling time to")
+            r_t=m.r_t,
+            doubling_time_t=abs(m.doubling_time_t),
+            impact_statement=("halves the infections every" if m.r_t < 1 else "reduces the doubling time to")
         )
     )
 

tests/test_app.py

@@ -76,7 +76,7 @@ def test_the_rest_of_header_shows_up():
 def test_mitigation_statement():
     st.cleanup()
     expected_doubling = "outbreak **reduces the doubling time to 7.8** days"
-    display_header(st, PARAM)
+    display_header(st, MODEL, PARAM)
     assert len((list(filter(lambda s: expected_doubling in s, st.render_store))))
     st.cleanup()
     expected_halving = "outbreak **halves the infections every 51.9** days"
@@ -92,7 +92,7 @@ def test_mitigation_statement():
         ventilated=RateLos(0.01, 10),
         n_days=60,
     )
-    display_header(st, halving_params)
+    display_header(st, MODEL, halving_params)
     assert len((list(filter(lambda s: expected_halving in s, st.render_store))))
     st.cleanup()