(original) (raw)

import array import io import sys import time LOOPS = 100000 INITIAL = "initial value" MORE = "more data" class StringBuilder(object): """Use it instead of doing += for building unicode strings from pieces""" def __init__(self, val=""): self.val = val self.appended = [] def __iadd__(self, other): self.appended.append(other) return self def __str__(self): self.val += "".join(self.appended) self.appended = [] return self.val def main_pure(loops): "str += str" b = INITIAL for i in range(loops): b += MORE return b def main_list_append(loops): "''.join()" b = [INITIAL] for i in range(loops): b.append(MORE) return "".join(b) def main_string_builder(loops): "StringBuilder" b = StringBuilder(INITIAL) for i in range(loops): b += MORE return str(b) def main_stringio(loops): "StringIO.join" b = io.StringIO(INITIAL) for i in range(loops): b.write(MORE) return b.getvalue() def main_array(loops): "array('u')" b = array.array('u', INITIAL) for i in range(loops): b.extend(MORE) return b.tounicode() ver = sys.version_info print("Python %s.%s" % (ver.major, ver.minor)) funcs = (main_pure, main_list_append, main_stringio, main_string_builder, main_array) width = 1 + max(len(func.__doc__) for func in funcs) for func in funcs: a = time.time() func(LOOPS) b = time.time() dt = b - a print("%s: %.0f ms" % (func.__doc__.ljust(width), dt * 1000))