from attractor import Attractor
from nose import with_setup
import numpy as np
def test_dt():
""" Our first test will be to ensure that our time increment, dt, is being properly calculated.
If this test fails we will be warned that there was a calculation error."""
a = Attractor()
assert a.end == a.dt * a.points, "\n There was a problem calculating the time increment! \n"
def test_euler():
""" This test case is to ensure proper size of the value returned by our euler method.
If this test fails it will indicate that there aren't the correct number of items being passed
to the method."""
a = Attractor()
xyz = np.array([0.0,0.0,0.0])
assert a.euler(xyz).shape == (3, ), "\n The array given to the euler method did not contain 3 items!\n"
def test_evolve():
""" This test case is to ensure proper number of parameters are being passed to the evolve method. As
above, this test also check for proper passing of paramters."""
a = Attractor()
assert a.evolve().shape == (a.points, 4), "\n The array given to the evolve method was improper!\n"
