for_while.py

#!/usr/bin/python
#coding: utf-8

import timeit

test1 = """
import itertools
def test1():
    count = itertools.count()
    for i in count:
        if i == 10000:
            return
"""
test2 = """
def test2():
    i = 0
    while True:
        if i == 10000:
            return
        i += 1
"""
if __name__ == '__main__':
    print timeit.timeit("test1()", test1, number=5000)
    # 1.53738689423
    print timeit.timeit("test2()", test2, number=5000)
    # 3.19617486