Changes in / [ba2ed54:7a3f840] in sasmodels

File:

• sasmodels/model_test.py (modified) (3 diffs)

sasmodels/model_test.py

@@ -47 +47 @@
 import sys
 import unittest
+
+try:
+    from StringIO import StringIO
+except ImportError: # StringIO.StringIO renamed to io.StringIO in Python 3
+    from io import StringIO
 
 import numpy as np  # type: ignore
@@ -350 +355 @@
 
     # Build a object to capture and print the test results
-    stream = _WritelnDecorator(sys.stdout)  # Add writeln() method to stream
+    stream = _WritelnDecorator(StringIO())  # Add writeln() method to stream
     verbosity = 2
     descriptions = True
@@ -388 +393 @@
     else:
         stream.writeln("Note: no test suite created --- this should never happen")
+
+    output = stream.getvalue()
+    stream.close()
+    return output