Splits the output of a pytest file into the format: filename.py 5 failures

parse_pytest.py

import re
import sys

start = re.compile('collected [0-9]+ items')
end = re.compile('=+ FAILURES =+')
fail = re.compile('F')
error = re.compile('E')

def split_failures(file_name):
    with open(file_name) as file_handler:
        contents = file_handler.read()
    start_match = start.search(contents)
    end_match = end.search(contents)
    return contents[start_match.end():end_match.start()].strip().split('\n')

def count_files(failures):
    for failure in failures:
        split_failure = failure.split(' ')
        num_failures = len(fail.findall(split_failure[1]))
        num_errors = len(error.findall(split_failure[1]))
        if num_failures > 0 or num_errors > 0:
            print "{}\n{} failures {} errors".format(split_failure[0], num_failures, num_errors)
        
def main():
    file_name = sys.argv[1]
    
    count_files(split_failures(file_name))

if __name__ == '__main__':
    main()