Changeset 9bc4882 in sasmodels for sasmodels/model_test.py

Timestamp:

Sep 5, 2017 11:01:03 AM (7 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, magnetic_model, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

8db25bf

Parents:

acc6c8f (diff), 142a8e2 (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge branch 'master' into ticket-776-orientation

File:

• sasmodels/model_test.py (modified) (4 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
+    # Note: io.StringIO exists in python 2, but using unicode instead of str
+    from io import StringIO
 
 import numpy as np  # type: ignore
@@ -337 +344 @@
 
 def run_one(model):
-    # type: (str) -> None
+    # type: (str) -> str
     """
     Run the tests for a single model, printing the results to stdout.
@@ -350 +357 @@
 
     # 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 +395 @@
     else:
         stream.writeln("Note: no test suite created --- this should never happen")
+
+    output = stream.getvalue()
+    stream.close()
+    return output