Added tests

Author: rumackaaron

changehc/tests/test_load_data.py

@@ -14,10 +14,12 @@
     "indicator": {
         "input_denom_file": "test_data/20200601_Counts_Products_Denom.dat.gz",
         "input_covid_file": "test_data/20200601_Counts_Products_Covid.dat.gz",
+        "input_flu_file": "test_data/20200601_Counts_Products_Covid.dat.gz",
         "drop_date": "2020-06-01"
     }
 }
 COVID_FILEPATH = PARAMS["indicator"]["input_covid_file"]
+FLU_FILEPATH = PARAMS["indicator"]["input_flu_file"]
 DENOM_FILEPATH = PARAMS["indicator"]["input_denom_file"]
 DROP_DATE = pd.to_datetime(PARAMS["indicator"]["drop_date"])
 
@@ -29,6 +31,8 @@ class TestLoadData:
                     Config.COVID_COLS, Config.COVID_DTYPES, Config.COVID_COL)
     combined_data = load_combined_data(DENOM_FILEPATH, COVID_FILEPATH, DROP_DATE,
                                             "fips")
+    flu_data = load_flu_data(DENOM_FILEPATH, FLU_FILEPATH, DROP_DATE,
+                                            "fips")
     gmpr = GeoMapper()
 
     def test_base_unit(self):
@@ -43,6 +47,9 @@ def test_base_unit(self):
         with pytest.raises(AssertionError):
             load_combined_data(DENOM_FILEPATH, COVID_FILEPATH, DROP_DATE, "foo")
 
+        with pytest.raises(AssertionError):
+            load_flu_data(DENOM_FILEPATH, FLU_FILEPATH, DROP_DATE, "foo")
+
     def test_denom_columns(self):
         assert "fips" in self.denom_data.index.names
         assert "timestamp" in self.denom_data.index.names
@@ -71,6 +78,16 @@ def test_combined_columns(self):
         assert len(
             set(self.combined_data.columns) - set(expected_combined_columns)) == 0
 
+    def test_flu_columns(self):
+        assert "fips" in self.flu_data.index.names
+        assert "timestamp" in self.flu_data.index.names
+
+        expected_flu_columns = ["num", "den"]
+        for col in expected_flu_columns:
+            assert col in self.flu_data.columns
+        assert len(
+            set(self.flu_data.columns) - set(expected_flu_columns)) == 0
+
     def test_edge_values(self):
         for data in [self.denom_data,
                      self.covid_data,