oo.py

import timeit

# sum the numbers from 0 to n-1 in different ways 


def while_loop(n=100_000_000):
    i = 0
    s = 0
    while i < n:
        s += i
        i += 1
    return s


def for_loop(n=100_000_000):
    s = 0
    for i in range(n):
        s += i
    return s

# -------------------------

def for_loop_increment(n=100_000_000):
    s = 0
    for i in range(n):
        s += 1
        i += 1
    return s


def for_loop_whit_test(n=100_000_000):
    s = 0
    for i in range(n):
        if i < n: pass
        s += i
    return s


def for_loop_with_increment_and_test(n=100_000_000):
    s = 0
    for i in range(n):
        if i < n: pass
        i += 1
        s += i
    return s 


def sum_range(n=100_000_000):
    return sum(range(n))


#def sum_numpy(n=100_000_000):
#    return numpy.sum(numpy.arange(n)) # numpy


def sum_math(n=100_000_000):
    return (n * (n - 1)) // 2

def main():
    print('sum math\t\t', timeit.timeit(sum_math, number=1))
    print('sum range\t\t', timeit.timeit(sum_range, number=1))
    print('while loop\t\t', timeit.timeit(while_loop, number=1))
    print('for loop\t\t', timeit.timeit(for_loop, number=1))
    print('for inc\t\t\t', timeit.timeit(for_loop_increment, number=1))
    print('for test\t\t', timeit.timeit(for_loop_whit_test, number=1))
    print('for inc+test\t\t', timeit.timeit(for_loop_with_increment_and_test, number=1))



if __name__ == '__main__':
    main()