Add CAN to JHU-to-S3 test

covid19-etl/tests/jhu_to_s3_test_data/time_series/country/CAN.json

@@ -0,0 +1 @@
+{"2020-01-22":{"confirmed":"<5","deaths":"<5","recovered":"<5"},"2020-03-12":{"confirmed":141,"deaths":"<5","recovered":8},"2020-05-01":{"confirmed":58549,"deaths":4381,"recovered":22764},"2020-06-20":{"confirmed":102652,"deaths":8609,"recovered":64826},"2020-08-09":{"confirmed":121296,"deaths":9067,"recovered":105420},"2020-09-28":{"confirmed":158510,"deaths":9360,"recovered":134551},"2020-11-17":{"confirmed":311043,"deaths":11220,"recovered":248859},"2021-01-06":{"confirmed":634437,"deaths":16539,"recovered":537024},"2021-02-25":{"confirmed":863301,"deaths":21868,"recovered":811551},"2021-04-16":{"confirmed":1115036,"deaths":23542,"recovered":1006819},"2021-06-05":{"confirmed":1399231,"deaths":25713,"recovered":1348019},"2021-07-25":{"confirmed":1433792,"deaths":26549,"recovered":1402035},"2021-09-13":{"confirmed":1559129,"deaths":27289,"recovered":"<5"},"2021-11-02":{"confirmed":1728543,"deaths":29110,"recovered":"<5"},"2021-12-22":{"confirmed":1935823,"deaths":30167,"recovered":"<5"}}

covid19-etl/tests/test_jhu_to_s3.py

@@ -98,10 +98,11 @@ def test_jhu_to_s3():
     assert len(all_locations["county"]) == len(uploaded_locations["county"])
 
     # check that the ETL generated the expected data.
-    # we only check a few locations: AFG (use case without state/
-    # county data) and USA (use case with state/county data) countries, and IL
-    # state and a few counties (because IL is the only state whose data we
-    # currently display on the frontend)
+    # we only check a few locations: AFG (use case without state/county data),
+    # CAN (use case with state data, without county data) and USA (use case
+    # with state/county data) countries, and IL state and a few counties
+    # (because IL is the only state whose data we currently display on the
+    # frontend)
     for data_level in ["country", "state", "county"]:
         dir = os.path.join(DATA_DIR, "time_series", data_level)
         for file in os.listdir(dir):