warn() goes to stderr instead of stdout, so print all messages

Author: nmdefries

src/acquisition/fluview/impute_missing_values.py

@@ -59,8 +59,6 @@
 from delphi.utils.geo.locations import Locations
 
 
-UNDERDETERMINED_MSG = "system is underdetermined"
-
 class Database:
     """Database wrapper and abstraction layer."""
 
@@ -243,7 +241,7 @@ def get_fusion_parameters(known_locations):
     H = graph[is_known, :]
     W = graph[is_unknown, :]
     if np.linalg.matrix_rank(H) != len(atoms):
-        raise StatespaceException(UNDERDETERMINED_MSG)
+        raise StatespaceException("system is underdetermined")
 
     HtH = np.dot(H.T, H)
     HtH_inv = np.linalg.inv(HtH)
@@ -299,11 +297,7 @@ def impute_missing_values(database, test_mode=False):
         try:
             F, known, unknown = get_fusion_parameters(known_values.keys())
         except StatespaceException as e:
-            message = str(e)
-            if message == UNDERDETERMINED_MSG:
-                warn(message)
-            else:
-                print(message)
+            print(e)
             continue
 
         # finally, impute the missing values

tests/acquisition/fluview/test_impute_missing_values.py

@@ -2,8 +2,10 @@
 
 # standard library
 import argparse
+import sys
 import unittest
 from unittest.mock import MagicMock
+from io import StringIO
 
 # first party
 from delphi.utils.geo.locations import Locations
@@ -86,8 +88,12 @@ 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.assertWarns(Warning):
-            impute_missing_values(db, test_mode=True)
+
+        capturedOutput = StringIO()
+        sys.stdout = capturedOutput
+        impute_missing_values(db, test_mode=True)
+        sys.stdout = sys.__stdout__
+        self.assertTrue("system is underdetermined" in capturedOutput.getvalue().split("\n"))
 
         db.find_missing_rows.return_value = [(201339, 201339)]
         impute_missing_values(db, test_mode=True)
@@ -130,5 +136,8 @@ 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.assertWarns(Warning):
-            impute_missing_values(db, test_mode=True)
+        capturedOutput = StringIO()
+        sys.stdout = capturedOutput
+        impute_missing_values(db, test_mode=True)
+        sys.stdout = sys.__stdout__
+        self.assertTrue("system is underdetermined" in capturedOutput.getvalue().split("\n"))