Merge pull request #667 from Rosuav/testfixes

A couple of small fixes to run-tests

1 files changed, 8 insertions, 7 deletions

diff --git a/tests/run-tests b/tests/run-tests
index 1e6dd50538..866e5a14cc 100755
--- a/tests/run-tests
+++ b/tests/run-tests
@@ -27,13 +27,17 @@ def run_tests(pyb, tests):
     failed_tests = []
     skipped_tests = []
 
-    running_under_travis = os.getenv('TRAVIS') == 'true'
+    skip_tests = set()
 
-    # Set of tests that we shouldn't run under Travis CI
-    skip_travis_tests = set(['basics/memoryerror.py'])
+    # Some tests shouldn't be run under Travis CI
+    if os.getenv('TRAVIS') == 'true':
+        skip_tests.add('basics/memoryerror.py')
 
     for test_file in tests:
-        if running_under_travis and test_file in skip_travis_tests:
+        test_basename = os.path.basename(test_file)
+        test_name = os.path.splitext(test_basename)[0]
+
+        if test_file in skip_tests:
             print("skip ", test_file)
             skipped_tests.append(test_name)
             continue
@@ -68,9 +72,6 @@ def run_tests(pyb, tests):
             except pyboard.PyboardError:
                 output_mupy = b'CRASH'
 
-        test_basename = os.path.basename(test_file)
-        test_name = os.path.splitext(test_basename)[0]
-
         if output_mupy == b'SKIP\n':
             print("skip ", test_file)
             skipped_tests.append(test_name)