Fixed ant_landscape/test.py so that it can be imported

This allows the `bl_load_py_modules.py` unit test to do its job.
This commit is contained in:
Sybren A. Stüvel 2017-06-19 15:11:46 +02:00
parent 7bd475449f
commit cb10969e46
1 changed files with 17 additions and 15 deletions

View File

@ -1,19 +1,21 @@
from stats import Stats
from numpy import *
stats = Stats()
a = zeros(10000000)
print(stats.time())
print(stats.memory())
a = sin(a)
print(stats.time())
print(stats.memory())
a = cos(a)
print(stats.time())
print(stats.memory())
a = cos(a)**2+sin(a)**2
print(stats.time())
print(stats.memory())
if __name__ == '__main__':
# This only works when this script is loaded as main, or
# run directly from the ant_landscape directory.
from stats import Stats
stats = Stats()
a = zeros(10000000)
print(stats.time())
print(stats.memory())
a = sin(a)
print(stats.time())
print(stats.memory())
a = cos(a)
print(stats.time())
print(stats.memory())
a = cos(a)**2+sin(a)**2
print(stats.time())
print(stats.memory())