tests to look for warnings instead

Author: nmdefries

tests/acquisition/fluview/test_impute_missing_values.py

@@ -11,7 +11,6 @@
     get_argument_parser,
     get_lag_and_ili,
     impute_missing_values,
-    StatespaceException,
 )
 
 # py3tester coverage target
@@ -87,7 +86,7 @@ def test_impute_missing_values_vipr(self):
         db.get_known_values.return_value = known_data
 
         db.find_missing_rows.return_value = [(201340, 201340)]
-        with self.assertRaises(Exception):
+        with self.assertWarns(Warning):
             impute_missing_values(db, test_mode=True)
 
         db.find_missing_rows.return_value = [(201339, 201339)]
@@ -131,5 +130,5 @@ def test_impute_missing_values_underdetermined(self):
         db.find_missing_rows.return_value = [(201740, 201740)]
         db.get_known_values.return_value = known_data
 
-        with self.assertRaises(StatespaceException):
+        with self.assertWarns(Warning):
             impute_missing_values(db, test_mode=True)