def test_generate_ngon():
"""
Checks if value that is passed to cos is in radians, i.e. cos(0) = 1
"""
temp = ChaosGame(4)
temp._generate_ngon()
nt.assert_equal(np.cos(temp.theta[0]), 1)
def test_savepng():
"""
Tests if a wrong format of the picture gives error
"""
temp = ChaosGame(2)
temp._generate_ngon()
temp._generate_points(20)
temp._starting_point()
temp.iterate(10)
a.savepng('filename.jpg')
def test_scaled():
"""
Test scaling vector - sum of the elements should be equal to 0
"""
temp = ChaosGame(4)
temp._generate_ngon()
temp._generate_points(20)
temp._starting_point()
temp.iterate(10)
computed = np.sum(temp.new_r[0])
assert (1 - computed) == 0
def test_generate_ngon():
""" Test that _generate_ngon returns list. """
T = ChaosGame(5, 0.5)
s = T._generate_ngon(5)
nt.assert_equal(type(s), list)
