update tests w/ new method arguments

Author: melange396

claims_hosp/tests/test_indicator.py

@@ -20,8 +20,8 @@
 
 
 class TestLoadData:
-    fips_data = load_data(DATA_FILEPATH, DROP_DATE, "fips")
-    hrr_data = load_data(DATA_FILEPATH, DROP_DATE, "hrr")
+    fips_data = load_data(DATA_FILEPATH, DROP_DATE, "fips", "Covid_like")
+    hrr_data = load_data(DATA_FILEPATH, DROP_DATE, "hrr", "Covid_like")
 
     def test_backwards_pad(self):
         num0 = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8], dtype=float).reshape(-1, 1)

claims_hosp/tests/test_load_data.py

@@ -21,15 +21,15 @@
 class TestLoadData:
     fips_claims_data = load_claims_data(DATA_FILEPATH, DROP_DATE, "fips")
     hrr_claims_data = load_claims_data(DATA_FILEPATH, DROP_DATE, "hrr")
-    fips_data = load_data(DATA_FILEPATH, DROP_DATE, "fips")
-    hrr_data = load_data(DATA_FILEPATH, DROP_DATE, "hrr")
+    fips_data = load_data(DATA_FILEPATH, DROP_DATE, "fips", "Covid_like")
+    hrr_data = load_data(DATA_FILEPATH, DROP_DATE, "hrr", "Covid_like")
 
     def test_base_unit(self):
         with pytest.raises(AssertionError):
             load_claims_data(DATA_FILEPATH, DROP_DATE, "foo")
 
         with pytest.raises(AssertionError):
-            load_data(DATA_FILEPATH, DROP_DATE, "foo")
+            load_data(DATA_FILEPATH, DROP_DATE, "foo", "Covid_like")
 
     def test_claims_columns(self):
         assert "hrr" in self.hrr_claims_data.index.names

claims_hosp/tests/test_update_indicator.py

@@ -52,6 +52,7 @@ def test_shift_dates(self):
             self.weekday,
             self.write_se,
             Config.signal_name,
+            "Covid_like",
             TEST_LOGGER
         )
         ## Test init
@@ -74,6 +75,7 @@ def test_geo_reindex(self):
             self.weekday,
             self.write_se,
             Config.signal_name,
+            "Covid_like",
             TEST_LOGGER
         )
         updater.shift_dates()
@@ -93,6 +95,7 @@ def test_update_indicator(self):
                 self.weekday,
                 self.write_se,
                 Config.signal_name,
+                "Covid_like",
                 TEST_LOGGER
             )
 
@@ -115,6 +118,7 @@ def test_write_to_csv_results(self):
             self.weekday,
             self.write_se,
             Config.signal_name,
+            "Covid_like",
             TEST_LOGGER
         )
 
@@ -196,6 +200,7 @@ def test_write_to_csv_with_se_results(self):
             True,
             True,
             signal_name,
+            "Covid_like",
             TEST_LOGGER
         )
 
@@ -248,6 +253,7 @@ def test_write_to_csv_wrong_results(self):
             self.weekday,
             self.write_se,
             Config.signal_name,
+            "Covid_like",
             TEST_LOGGER
         )