def test_move_limit_after_200_iterations(self): expected = [0.2000, 0.2000, 0.2000, 0.2000, 0.2000] p = [0, 1, 0, 0, 0] U = 1 for i in range(200): p = lt.xmove(p, U, 1, 0, 0.8, 0.1, 0.1) self.assertAlmostEqual(expected, p, places=4)
def test_xmove(self): expected = [0.0000, 0.1000, 0.8000, 0.1000, 0.0000] p = [0, 1, 0, 0, 0] U = 1 p = lt.xmove(p, U, 1, 1, 0.8, 0.1, 0.1) self.assertAlmostEqual(expected, p, places=4)