python pillow simple benchmark

main.py

import json
import os.path
import timeit
from collections import namedtuple

from PIL import Image

SizeUnit = namedtuple('SizeUnit', ['size', 'name'])
KB = SizeUnit(1024, 'KB')
MB = SizeUnit(1024*1024, 'MB')


def pretty_size(size, size_unit=KB):
    return '{:.3f} {}'.format(size / size_unit.size, size_unit.name)


def run(input_name, output_name, op, **save_kwargs):

    data = {
        'input_name': input_name,
        'output_name': output_name,
        'op': output_name,
        'save_kwargs': save_kwargs,
    }

    def _run():
        pre_size = os.path.getsize(input_name)

        img = Image.open(input_name)
        img = op(img)
        img.save(output_name, **save_kwargs)

        new_size = os.path.getsize(output_name)

        data['input_size'] = pretty_size(pre_size, MB)
        data['output_size'] = pretty_size(new_size, KB)

    data['time'] = '{:.4f} s'.format(timeit.timeit(_run, number=1))
    print(json.dumps(data, indent='\t'))


def thumbnail(img: Image):
    img.thumbnail((250, 250))
    return img


def noop(img):
    return img


run('input.jpg', 'thumbnail.jpg', thumbnail)
run('input.jpg', 'quality30.jpg', noop, quality=30)