def test_burnin(self): v = np.array([1, 2]) nb_its = 10 pace = 1 start = 5 t = Trajectory(v, ['time'], {'time':np.array([1,2])}, pace, start, nb_its, first_saved_iteration=start) for i in xrange(nb_its): v *= 2 t.record(i) assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its) self.assertEqual(t.saved_iterations, range(start,nb_its))
def test_basic(self): v = np.array([1, 2]) nb_its = 10 pace = 1 start = 0 t = Trajectory(v, ['time'], {'time':np.array([1,2])}, pace, start, nb_its) for i in xrange(nb_its): v *= 2 t.increment(i) assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its) self.assertEqual(t.saved_iterations, [-1] + range(start,nb_its))
def test_basic(self): v = np.array([1, 2]) nb_its = 10 pace = 1 start = 0 t = Trajectory(v, ['time'], {'time': np.array([1, 2])}, pace, start, nb_its) for i in xrange(nb_its): v *= 2 t.record(i) assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its) self.assertEqual(t.saved_iterations, [-1] + range(start, nb_its))