spiff-arena/SpiffWorkflow/scripts/test_times.py

#!/usr/bin/env python

# Copyright (C) 2023 Sartography
#
# This file is part of SpiffWorkflow.
#
# SpiffWorkflow is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 3.0 of the License, or (at your option) any later version.
#
# SpiffWorkflow is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301  USA

import re
import sys

def regex_line_parser(pattern, handler):
    regex = re.compile(pattern)
    def parser(line):
        match = regex.match(line)
        if match:
            return handler(match)
        return None
    return parser

def rstripped(match):
    return match.group(0).rstrip()

def tupled(match):
    return (match.group(1), match.group(2))


def parse(lines):
    test_file = None
    timing = None
    test_file_timings = []

    test_file_line_parser = regex_line_parser('.*?Test.py', rstripped)
    timing_line_parser = regex_line_parser('Ran (.*) tests? in (.*)', tupled)

    for line in lines:
        if test_file is None:
            test_file = test_file_line_parser(line)
        elif timing is None:
            timing = timing_line_parser(line)

        if test_file is not None and timing is not None:
            test_file_timings.append((test_file, timing))
            test_file = None
            timing = None

    return test_file_timings

def report(parsed_data):
    lines = [
        '| Method | Time | Tests Ran |',
        '|----|----|----|',
    ]

    sorted_data = sorted(parsed_data, key=lambda d: d[1][1], reverse=True)
    for d in sorted_data:
        lines.append(f'| {d[0]} | {d[1][1]} | {d[1][0]} |')

    print('\n'.join(lines))

if __name__ == '__main__':
    data = sys.stdin.readlines()
    parsed_data = parse(data)
    report(parsed_data)