import os import subprocess import re import sys try: import xmlrunner except: print "xmlrunner needs to be installed to run these tests" print "Try easy_install unittest-xml-reporting" # Check whether we have matplotlib installed HAS_MPL_WX = True try: import matplotlib import wx except: HAS_MPL_WX = False SKIPPED_DIRS = ["sansrealspace", "calculatorview"] if not HAS_MPL_WX: SKIPPED_DIRS.append("sansguiframe") COMMAND_SEP = ';' if os.name == 'nt': COMMAND_SEP = '&' def run_tests(): passed = 0 failed = 0 n_tests = 0 n_errors = 0 n_failures = 0 for d in os.listdir(os.getcwd()): # Check for modules to be skipped if d in SKIPPED_DIRS: continue # Go through modules looking for unit tests module_dir = os.path.join(os.getcwd(),d,"test") if os.path.isdir(module_dir): for f in os.listdir(module_dir): file_path = os.path.join(module_dir,f) if os.path.isfile(file_path) and f.startswith("utest_") and f.endswith(".py"): module_name,_ = os.path.splitext(f) code = "cd %s%s%s -c \"import sys;import xmlrunner;import unittest;sys.path.insert(0, '%s');" % (module_dir, COMMAND_SEP, sys.executable, module_dir) code += "from %s import *;" % module_name code += "unittest.main(testRunner=xmlrunner.XMLTestRunner(output='logs'))\"" proc = subprocess.Popen(code, shell=True, stdout=subprocess.PIPE, stderr = subprocess.STDOUT) std_out, std_err = proc.communicate() has_failed = True m = re.search("Ran ([0-9]+) test", std_out) if m is not None: has_failed = False n_tests += int(m.group(1)) m = re.search("FAILED \(errors=([0-9]+)\)", std_out) if m is not None: has_failed = True n_errors += int(m.group(1)) m = re.search("FAILED \(failures=([0-9]+)\)", std_out) if m is not None: has_failed = True n_failures += int(m.group(1)) if has_failed: failed += 1 print "Result for %s (%s): FAILED" % (module_name, module_dir) print std_out else: passed += 1 print "Result for %s: SUCCESS" % module_name print "\n----------------------------------------------" print "Results by test modules:" print " PASSED: %d" % passed ratio = 100.0*failed/(failed+passed) print " FAILED: %d (%2.2g%%)" % (failed,ratio) print "Results by tests:" print " Tests run: %d" % n_tests print " Tests failed: %d" % n_failures print " Test errors: %d" % n_errors print "----------------------------------------------" return failed if __name__ == '__main__': if run_tests()>0: sys.exit(1)