python string replace benchmark

replace-whitespace.py

import time
import re
import string

src = '_ ' * 2**20 # 2MB


whitespacere = re.compile('\s')

def replace():
    return src.replace(' ', '').replace('\r', '').replace('\n', '').replace('\t', '')


def sub():
    return whitespacere.sub('', src)

def translate():
    return src.translate(None, string.whitespace)


def testrun(f, i=500):
    start = time.time()
    for _ in range(i):
        f()
    end = time.time()
    print("{} for {} runs of {}".format(
            end - start,
            i,
            f.__name__
        ))



if __name__ == '__main__':
    testrun(translate)
    testrun(replace)
    testrun(sub)