Fixes for downstream applications that use Python 2

sciunit/utils.py

@@ -11,6 +11,7 @@
 import importlib
 import json
 import re
+import contextlib
 from io import TextIOWrapper, StringIO
 from datetime import datetime
 
@@ -71,6 +72,17 @@ def printd(*args, **kwargs):
     return False
 
 
+if PYTHON_MAJOR_VERSION == 3:
+    redirect_stdout = contextlib.redirect_stdout
+else:  # Python 2
+    @contextlib.contextmanager
+    def redirect_stdout(target):
+        original = sys.stdout
+        sys.stdout = target
+        yield
+        sys.stdout = original
+
+
 def assert_dimensionless(value):
     """
     Tests for dimensionlessness of input.