benchmark-io.py

"""Benchmark IO cross Python2 and Python3.

Result:

    1. Python2 + cString
    2. Python3 + six.bytes.io
    3. Python2 + six.bytes.io

Log:

    Python3 six bytes io
    running io benchmark on Python3
    for 20000 loops, cost 0.015699999999999992

    Python2 six bytes io
    running io benchmark on Python2
    for 20000 loops, cost 0.030913

    Python2 cString IO
    running io benchmark on Python2
    for 20000 loops, cost 0.01276

"""

from __future__ import print_function

import sys
import time

import six

if six.PY3:
    timer_func = time.process_time
else:
    timer_func = time.clock

if six.PY3:
    from six import BytesIO as IO
else:
    from cStringIO import StringIO as IO
    # from six import BytesIO as IO

io_func = IO
io_obj = io_func


def benchmark():
    print("running io benchmark on Python{}".format(str(sys.version[0])))

    n = 20000
    start = timer_func()
    for i in range(n):
        io_func(bytes(i))

    end = timer_func()
    print("for {} loops, cost {}".format(str(n), str(end-start)))


benchmark()