Merge pull request #180 from CodeForPhilly/179-chart-tests

179 chart tests

Author: quinn-dougherty

tests/__init__.py

@@ -0,0 +1,62 @@
+,day,hosp,icu,vent
+0,0,,,
+1,1,1.0,1.0,1.0
+2,2,2.0,1.0,1.0
+3,3,3.0,1.0,1.0
+4,4,4.0,2.0,1.0
+5,5,5.0,2.0,1.0
+6,6,6.0,3.0,2.0
+7,7,8.0,3.0,2.0
+8,8,9.0,4.0,2.0
+9,9,10.0,5.0,3.0
+10,10,11.0,5.0,3.0
+11,11,12.0,6.0,3.0
+12,12,14.0,7.0,4.0
+13,13,15.0,7.0,4.0
+14,14,17.0,8.0,5.0
+15,15,19.0,9.0,5.0
+16,16,22.0,10.0,6.0
+17,17,24.0,12.0,6.0
+18,18,27.0,13.0,7.0
+19,19,30.0,14.0,8.0
+20,20,34.0,16.0,9.0
+21,21,38.0,18.0,10.0
+22,22,42.0,20.0,11.0
+23,23,48.0,22.0,12.0
+24,24,53.0,25.0,14.0
+25,25,60.0,28.0,15.0
+26,26,67.0,31.0,17.0
+27,27,75.0,35.0,19.0
+28,28,84.0,39.0,21.0
+29,29,94.0,44.0,24.0
+30,30,105.0,49.0,26.0
+31,31,118.0,55.0,29.0
+32,32,132.0,62.0,33.0
+33,33,147.0,69.0,37.0
+34,34,165.0,77.0,41.0
+35,35,184.0,86.0,46.0
+36,36,206.0,96.0,51.0
+37,37,230.0,108.0,57.0
+38,38,256.0,120.0,64.0
+39,39,286.0,134.0,71.0
+40,40,319.0,149.0,80.0
+41,41,355.0,167.0,89.0
+42,42,395.0,185.0,99.0
+43,43,439.0,206.0,110.0
+44,44,488.0,229.0,122.0
+45,45,541.0,254.0,135.0
+46,46,599.0,282.0,150.0
+47,47,662.0,312.0,166.0
+48,48,731.0,345.0,184.0
+49,49,806.0,380.0,203.0
+50,50,886.0,418.0,223.0
+51,51,971.0,460.0,
+52,52,1062.0,,
+53,53,1159.0,,
+54,54,,,
+55,55,,,
+56,56,,,
+57,57,,,
+58,58,,,
+59,59,,,
+60,60,,,

tests/census_df.csv

@@ -0,0 +1,62 @@
+,day,hosp,icu,vent
+0,0,,,
+1,1,0.7347722898562381,0.29390891594249524,0.14695445797124762
+2,2,0.8247048404412878,0.3298819361765153,0.16494096808825764
+3,3,0.925631917053221,0.37025276682128805,0.18512638341064402
+4,4,1.0388943119331255,0.4155577247732505,0.20777886238662524
+5,5,1.1659954645505692,0.4663981858202275,0.23319909291011376
+6,6,1.3086209670885331,0.5234483868354127,0.26172419341770636
+7,7,1.4686603504707918,0.5874641401883167,0.29373207009415836
+8,8,1.6482314021401603,0.6592925608560645,0.32964628042803223
+9,9,1.8497072903287552,0.7398829161315037,0.36994145806575185
+10,10,2.075746794129696,0.8302987176518766,0.4151493588259383
+11,11,2.329327963945971,0.9317311855783883,0.46586559278919415
+12,12,2.61378556236113,1.0455142249444513,0.5227571124722257
+13,13,2.932852660451122,1.1731410641804523,0.5865705320902261
+14,14,3.2907067880796568,1.3162827152318588,0.6581413576159294
+15,15,3.692021057497506,1.476808422999003,0.7384042114995015
+16,16,4.142020695886615,1.6568082783546476,0.8284041391773238
+17,17,4.646545432100844,1.8586181728403375,0.9293090864201687
+18,18,5.212118182837166,2.0848472731348657,1.0424236365674329
+19,19,5.846020470068794,2.3384081880275147,1.1692040940137574
+20,20,6.556374969989172,2.622549987995672,1.311274993997836
+21,21,7.352235537885861,2.9408942151543442,1.4704471075771721
+22,22,8.243684965656087,3.2974739862624354,1.6487369931312177
+23,23,9.24194059957226,3.6967762398288997,1.8483881199144498
+24,24,10.35946776360953,4.143787105443813,2.0718935527219067
+25,25,11.610100683671433,4.644040273468569,2.3220201367342845
+26,26,13.009170272720908,5.203668109088369,2.6018340545441845
+27,27,14.57363769481826,5.8294550779272996,2.9147275389636498
+28,28,16.32223205189976,6.528892820759907,3.2644464103799535
+29,29,18.275589800725783,7.310235920290317,3.6551179601451587
+30,30,20.45639257382453,8.182557029529804,4.091278514764902
+31,31,22.889498907537188,9.155799563014867,4.577899781507433
+32,32,25.602063927916674,10.240825571166681,5.120412785583341
+33,33,28.623639263122044,11.449455705248809,5.7247278526244045
+34,34,31.986243289314473,12.794497315725806,6.397248657862903
+35,35,35.72438922785608,14.289755691142432,7.144877845571216
+36,36,39.875055553763275,15.950022221505293,7.9750111107526465
+37,37,44.47757962283373,17.79103184913353,8.895515924566766
+38,38,49.573451377900426,19.829380551160142,9.914690275580071
+39,39,55.20597949586033,22.082391798344105,11.041195899172052
+40,40,61.41979749256973,24.56791899702793,12.283959498513966
+41,41,68.26017230956847,27.304068923827344,13.652034461913672
+42,42,75.77207308773609,30.308829235094493,15.154414617547246
+43,43,83.9989536775131,33.599581471005195,16.799790735502597
+44,44,92.98119964480816,37.192479857923274,18.596239928961637
+45,45,102.75419006634718,41.10167602653888,20.55083801326944
+46,46,113.34592752723552,45.33837101089421,22.669185505447103
+47,47,124.77419801601036,49.90967920640412,24.95483960320206
+48,48,137.04323774220597,54.81729509688239,27.408647548441195
+49,49,150.13990838845984,60.05596335538394,30.02798167769197
+50,50,164.02941811859068,65.61176724743632,32.80588362371816
+51,51,178.65067471298744,71.46026988519498,35.73013494259749
+52,52,193.91142069932403,77.56456827972954,38.78228413986477
+53,53,209.6833781349851,83.87335125399409,41.936675626997044
+54,54,225.79772044013316,90.31908817605324,45.15954408802662
+55,55,242.04128493041298,96.81651397216513,48.40825698608256
+56,56,258.15403301982815,103.2616132079313,51.63080660396565
+57,57,273.828341452579,109.53133658103161,54.765668290515805
+58,58,288.71074888939893,115.48429955575966,57.74214977787983
+59,59,302.40676593155194,120.96270637262069,60.48135318631034
+60,60,,,

tests/projection_admits.csv

@@ -3,14 +3,15 @@
 import pytest  # type: ignore
 import pandas as pd  # type: ignore
 import numpy as np  # type: ignore
+import altair as alt  # type: ignore
+
+from src.penn_chime.charts import new_admissions_chart, admitted_patients_chart
+from src.penn_chime.models import sir, sim_sir
+from src.penn_chime.parameters import Parameters
+from src.penn_chime.presentation import display_header
+from src.penn_chime.settings import DEFAULTS
+from src.penn_chime.defaults import RateLos
 
-from app import (admissions_df, alt)
-from penn_chime.models import sir, sim_sir, sim_sir_df
-from penn_chime.parameters import Parameters
-from penn_chime.presentation import display_header
-from penn_chime.charts import new_admissions_chart
-from penn_chime.settings import DEFAULTS
-from penn_chime.defaults import RateLos
 PARAM = Parameters(
         current_hospitalized=100,
         doubling_time=6.0,
@@ -24,6 +25,7 @@
         n_days=60
     )
 
+
 # set up
 
 # we just want to verify that st _attempted_ to render the right stuff
@@ -157,35 +159,36 @@ def test_sim_sir():
         assert isinstance(v, np.ndarray)
 
 
-## THIS function never gets called in the app so i'm commenting out its test
-#ef test_sim_sir_df():
-#   """
-#   Rounding to move fast past decimal place issues
-#   """
-#
-#   df = sim_sir_df(PARAM)
-#   first = df.iloc[0]
-#   last = df.iloc[-1]
-#   assert round(first[0], 0) == 5
-#   assert round(first[1], 2) == 6
-#   assert round(first[2], 0) == 7
-#   assert round(last[0], 2) == 0
-#   assert round(last[1], 2) == 0.18
-#   assert round(last[2], 2) == 17.82
-
-
 def test_new_admissions_chart():
-    chart = new_admissions_chart(alt, admissions_df, PARAM)
+    projection_admits = pd.read_csv('tests/projection_admits.csv')
+    chart = new_admissions_chart(alt, projection_admits, PARAM)
     assert isinstance(chart, alt.Chart)
-    assert chart.data.iloc[1].Hospitalized < 1
-    # assert round(chart.data.iloc[49].ICU, 0) == 43
+    assert chart.data.iloc[1].hosp < 1
+    assert round(chart.data.iloc[40].icu, 0) == 25
+
+    # test fx call with no params
     with pytest.raises(TypeError):
         new_admissions_chart()
 
     empty_chart = new_admissions_chart(alt, pd.DataFrame(), PARAM)
     assert empty_chart.data.empty
 
 
+def test_admitted_patients_chart():
+    census_df = pd.read_csv('tests/census_df.csv')
+    chart = admitted_patients_chart(alt, census_df, PARAM)
+    assert isinstance(chart, alt.Chart)
+    assert chart.data.iloc[1].hosp == 1
+    assert chart.data.iloc[49].vent == 203
+
+    # test fx call with no params
+    with pytest.raises(TypeError):
+        admitted_patients_chart()
+
+    empty_chart = admitted_patients_chart(alt, pd.DataFrame(), PARAM)
+    assert empty_chart.data.empty
+
+
 def test_parameters():
     param = Parameters(
         current_hospitalized=100,