Updated circle_tests.py to run either py2 or py3 tests from config.yml

Author: cmonr

circle_tests.py

@@ -26,27 +26,51 @@ def remove_readonly(func, path, _):
 
     shutil.rmtree(directory, onerror=remove_readonly)
 
+# Source tests from yaml config
 tests = None
-with open("circle.yml", "r") as f:
-    types = yaml.load_all(f)
-    for t in types:
-        for k,v in t.items():
-            if k == 'test':
-                tests = v['override']
-
-if tests:
-    cwd = os.path.abspath(os.path.dirname(__file__))
-
-    if os.path.exists(os.path.join(cwd, '.tests')):
-        rmtree_readonly(os.path.join(cwd, '.tests'))
-    os.mkdir(os.path.join(cwd, '.tests'))
-
-    for cmd in tests:
-        os.chdir(cwd)
-        print "\n----------\nEXEC: \"%s\" " % cmd
-        proc = subprocess.Popen(cmd, shell=True)
-        proc.communicate()
-
-        if proc.returncode != 0:
-            print "\n------------\nERROR: \"%s\"" % cmd
-            sys.exit(1)
+with open('.circleci/config.yml', 'r') as f:
+    data = yaml.safe_load(f)
+
+    # Read yaml tree
+    if sys.version_info[0] == 3:
+        tests = data['jobs']['py3']['steps']
+    else:
+        tests = data['jobs']['py2']['steps']
+
+    # Filter command list to only contain commands
+    tests = [item['run'] for item in list(filter(lambda x : type(x) is dict, tests))]
+
+    # ... and replace new lines with ampersands
+    tests = [item.replace('\n', ' && ') for item in tests]
+
+# Exit if no tests are found
+if tests == None:
+    sys.exit(1)
+
+# Ignore all tests found before `pip install -e`
+startIndex = -1
+for cmd in tests:
+    if 'pip install -e' in cmd:
+        startIndex = tests.index(cmd) + 1
+        break
+
+if startIndex == -1:
+    sys.exit(1)
+
+# Delete `.test` directory if it exists
+cwd = os.path.abspath(os.path.dirname(__file__))
+
+if os.path.exists(os.path.join(cwd, '.tests')):
+    rmtree_readonly(os.path.join(cwd, '.tests'))
+os.mkdir(os.path.join(cwd, '.tests'))
+
+# Run tests
+for cmd in tests[startIndex:]:
+    os.chdir(cwd)
+    print("\n----------\nEXEC: \"%s\" " % cmd)
+    proc = subprocess.Popen(cmd, shell=True)
+    proc.communicate()
+
+    if proc.returncode != 0:
+        print "\n------------\nERROR: \"%s\"" % cmd
+        sys.exit(1)