Port public R queries

Author: rzats

tests/test_epidata_calls.py

@@ -0,0 +1,174 @@
+from epidatpy.request import Epidata, EpiRange
+
+# Requirements to run these:
+# DELPHI_EPIDATA_KEY environment variable is set https://api.delphi.cmu.edu/epidata/admin/registration_form
+
+def test_pub_covid_hosp_facility_lookup() -> None:
+    apicall = Epidata.pub_covid_hosp_facility_lookup(state="fl")
+    data = apicall.json()
+    assert(len(data) > 0)
+
+    apicall = Epidata.pub_covid_hosp_facility_lookup(city="southlake")
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_covid_hosp_facility() -> None:
+    apicall = Epidata.pub_covid_hosp_facility(
+        hospital_pks = "100075",
+        collection_weeks = EpiRange(20200101, 20200501))
+    data = apicall.json()
+    assert(len(data) > 0)
+
+    apicall = Epidata.pub_covid_hosp_facility(
+        hospital_pks = "100075",
+        collection_weeks = EpiRange(202001, 202005))
+    data = apicall.json()
+    assert(len(data) > 0) # fails
+
+def test_pub_covid_hosp_state_timeseries() -> None:
+    apicall = Epidata.pub_covid_hosp_state_timeseries(
+        states = "fl",
+        dates = EpiRange(20200101, 20200501))
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_covidcast_meta() -> None:
+    apicall = Epidata.pub_covidcast_meta()
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_covidcast() -> None:
+    apicall = Epidata.pub_covidcast(
+        data_source = "jhu-csse",
+        signals = "confirmed_7dav_incidence_prop",
+        geo_type = "state",
+        time_type = "day",
+        geo_values = ["ca", "fl"],
+        time_values = EpiRange(20200601, 20200801))
+    data = apicall.json()
+    assert(len(data) > 0)
+
+    apicall = Epidata.pub_covidcast(
+        data_source = "jhu-csse",
+        signals = "confirmed_7dav_incidence_prop",
+        geo_type = "state",
+        time_type = "day",
+        geo_values = "*",
+        time_values = EpiRange(20200601, 20200801))
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_delphi() -> None:
+    apicall = Epidata.pub_delphi(
+        system = "ec",
+        epiweek = 201501
+    )
+    data = apicall.classic() # only supports classic
+    assert(len(data) > 0)
+
+def test_pub_dengue_nowcast() -> None:
+    apicall = Epidata.pub_dengue_nowcast(
+        locations = "pr",
+        epiweeks = EpiRange(201401, 202301)
+    )
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_ecdc_ili() -> None:
+    apicall = Epidata.pub_ecdc_ili(
+        regions = "austria",
+        epiweeks = EpiRange(201901, 202001)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_flusurv() -> None:
+    apicall = Epidata.pub_flusurv(
+        locations = "CA",
+        epiweeks = EpiRange(201701, 201801)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_fluview_clinical() -> None:
+    apicall = Epidata.pub_fluview_clinical(
+        regions = "nat",
+        epiweeks = EpiRange(201601, 201701)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_fluview_meta() -> None:
+    apicall = Epidata.pub_fluview_meta()
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_fluview() -> None:
+    apicall = Epidata.pub_fluview(
+        regions = "nat",
+        epiweeks = EpiRange(201201, 202005)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_gft() -> None:
+    apicall = Epidata.pub_gft(
+        locations = "hhs1",
+        epiweeks = EpiRange(201201, 202001)
+    )
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_kcdc_ili() -> None:
+    apicall = Epidata.pub_kcdc_ili(
+        regions = "ROK",
+        epiweeks = 200436
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_meta() -> None:
+    apicall = Epidata.pub_meta()
+    data = apicall.classic() # only supports classic
+    assert(len(data) > 0)
+
+def test_pub_nidss_dengue() -> None:
+    apicall = Epidata.pub_nidss_dengue(
+        locations = "taipei",
+        epiweeks = EpiRange(201201, 201301)
+    )
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_nidss_flu() -> None:
+    apicall = Epidata.pub_nidss_flu(
+        regions = "taipei",
+        epiweeks = EpiRange(201501, 201601)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_nowcast() -> None:
+    apicall = Epidata.pub_nowcast(
+        locations = "ca",
+        epiweeks = EpiRange(201201, 201301)
+    )
+    data = apicall.json()
+    assert(len(data) > 0)
+
+def test_pub_paho_dengue() -> None:
+    apicall = Epidata.pub_paho_dengue(
+        regions = "ca",
+        epiweeks = EpiRange(201401, 201501)
+    )
+    data = apicall.json(disable_date_parsing=True)
+    assert(len(data) > 0)
+
+def test_pub_wiki() -> None:
+    apicall = Epidata.pub_wiki(
+        articles = "avian_influenza",
+        time_type = "week",
+        time_values = EpiRange(201501, 201601)
+    )
+    data = apicall.json()
+    assert(len(data) > 0)