Benchmark for PyPy's numpypy

numpypy_simplebench.py

#Simple benchmark for PyPy's numpypy
#Based on https://bitbucket.org/pypy/pypy/src/dd3bcd84b145/pypy/module/micronumpy/bench/

import time

try:
    import numpypy as numpy
except:
    import numpy

def add():
    a = numpy.ones(10000000)
    a = a + a + a + a + a 
    a[0] = 3.0 

def iterate():
    sum = 0 
    a = numpy.ones(10000000)
    for i in range(10000000):
        sum += a[i]
    return sum 

total = 0.0 
for i in range(10):
        s = time.time()
        add()
        total += time.time() - s 
print "add :", total / 10.0, "sec"

total = 0.0 
for i in range(10):
        s = time.time()
        iterate()
        total += time.time() - s 
print "iterate :", total / 10.0, "sec"