Sort the times produced by pytest --durations=0

sort_pytest_durations.py

# coding: utf-8
"""Some helpful functions for neatly displaying the output of `pytest
--durations=0`."""
import itertools
import re
import sys

def parse_lines(text):
    results = []
    for line in text.splitlines():
        match = re.match(
            r'^(?P<time>\d+.\d+)s (?P<call_setup>call|setup)\s+'
            r'(?P<test_name>.*?)(\[(?P<params>.*)\])?\s*$', line)
        if match:
            match_dict = match.groupdict()
            new_dict = dict(
                time=float(match_dict['time']),
                ident=(
                    match_dict['test_name'],
                    match_dict['call_setup'] + '::' + (match_dict['params'] or '')))
            results.append(new_dict)
    return results

def sort_into_groups(results):
    key = lambda d: d['ident'][0]
    sorted_results = sorted(results, key=key)
    group_iter = itertools.groupby(sorted_results, key)
    results = []
    for group_ident, group in group_iter:
        group_list = list(group)
        group_time = sum(d['time'] for d in group_list)
        sub_idents = [(d['time'], d['ident'][1]) for d in group_list]
        results.append((group_time, group_ident, sub_idents))
    return sorted(results, reverse=True)

def print_sorted_results(sorted_results):
    for group_time, group_ident, group in sorted_results:
        print('\n% 7.2f %s' % (group_time, group_ident))
        for elem_time, elem_name in group:
            print('    % 4.2f %s' % (elem_time, elem_name))

data = sys.stdin.read()
results = parse_lines(data)
sorted_results = sort_into_groups(results)
print_sorted_results(sorted_results)