test: check if output is set

if exception is thrown before setting the output, this will catch it.
oldshuren · Feb 23, 2017 · 43b2585 · 43b2585
1 parent 5a585ee
commit 43b2585
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test/test_ulog2csv.py b/test/test_ulog2csv.py
@@ -21,18 +21,20 @@ def run_against_file(self, expected_output_file, test):
         saved_stdout = sys.stdout
         with open(expected_output_file, 'r') as file_handle:
             expected_output = file_handle.read().strip()
+            output = None
             try:
                 out = StringIO()
                 sys.stdout = out
                 test()
                 output = out.getvalue().strip()
                 assert output == expected_output
             finally:
-                sys.stdout = saved_stdout
-                print("Got output:")
-                print(output)
-                print("\nExpected output:")
-                print(expected_output)
+                if output is not None:
+                    sys.stdout = saved_stdout
+                    print("Got output:")
+                    print(output)
+                    print("\nExpected output:")
+                    print(expected_output)
 
     def test_ulog2csv(self):
         tmpdir = tempfile.gettempdir()