def test_approx_mdl_traj():
t = np.linspace(0, 1.75 * 2 * np.pi, 100)
x = np.sin(t)
y = np.cos(t)
z = t
xyz = np.vstack((x, y, z)).T
xyza1 = pf.approximate_mdl_trajectory(xyz, alpha=1.)
xyza2 = pf.approximate_mdl_trajectory(xyz, alpha=2.)
assert_equal(len(xyza1), 10)
assert_equal(len(xyza2), 8)
assert_array_almost_equal(
xyza1,
np.array([[0.00000000e+00, 1.00000000e+00, 0.00000000e+00],
[9.39692621e-01, 3.42020143e-01, 1.22173048e+00],
[6.42787610e-01, -7.66044443e-01, 2.44346095e+00],
[-5.00000000e-01, -8.66025404e-01, 3.66519143e+00],
[-9.84807753e-01, 1.73648178e-01, 4.88692191e+00],
[-1.73648178e-01, 9.84807753e-01, 6.10865238e+00],
[8.66025404e-01, 5.00000000e-01, 7.33038286e+00],
[7.66044443e-01, -6.42787610e-01, 8.55211333e+00],
[-3.42020143e-01, -9.39692621e-01, 9.77384381e+00],
[-1.00000000e+00, -4.28626380e-16, 1.09955743e+01]]))
assert_array_almost_equal(
xyza2,
np.array([[0.00000000e+00, 1.00000000e+00, 0.00000000e+00],
[9.95471923e-01, -9.50560433e-02, 1.66599610e+00],
[-1.89251244e-01, -9.81928697e-01, 3.33199221e+00],
[-9.59492974e-01, 2.81732557e-01, 4.99798831e+00],
[3.71662456e-01, 9.28367933e-01, 6.66398442e+00],
[8.88835449e-01, -4.58226522e-01, 8.32998052e+00],
[-5.40640817e-01, -8.41253533e-01, 9.99597663e+00],
[-1.00000000e+00, -4.28626380e-16, 1.09955743e+01]]))
def test_approx_mdl_traj():
t=np.linspace(0,1.75*2*np.pi,100)
x = np.sin(t)
y = np.cos(t)
z = t
xyz=np.vstack((x,y,z)).T
xyza1 = pf.approximate_mdl_trajectory(xyz,alpha=1.)
xyza2 = pf.approximate_mdl_trajectory(xyz,alpha=2.)
assert_equal(len(xyza1), 10)
assert_equal(len(xyza2), 8)
assert_array_almost_equal( xyza1, np.array([[ 0.00000000e+00, 1.00000000e+00, 0.00000000e+00],
[ 9.39692621e-01, 3.42020143e-01, 1.22173048e+00],
[ 6.42787610e-01, -7.66044443e-01, 2.44346095e+00],
[ -5.00000000e-01, -8.66025404e-01, 3.66519143e+00],
[ -9.84807753e-01, 1.73648178e-01, 4.88692191e+00],
[ -1.73648178e-01, 9.84807753e-01, 6.10865238e+00],
[ 8.66025404e-01, 5.00000000e-01, 7.33038286e+00],
[ 7.66044443e-01, -6.42787610e-01, 8.55211333e+00],
[ -3.42020143e-01, -9.39692621e-01, 9.77384381e+00],
[ -1.00000000e+00, -4.28626380e-16, 1.09955743e+01]]))
assert_array_almost_equal(xyza2, np.array([[ 0.00000000e+00, 1.00000000e+00, 0.00000000e+00],
[ 9.95471923e-01, -9.50560433e-02, 1.66599610e+00],
[ -1.89251244e-01, -9.81928697e-01, 3.33199221e+00],
[ -9.59492974e-01, 2.81732557e-01, 4.99798831e+00],
[ 3.71662456e-01, 9.28367933e-01, 6.66398442e+00],
[ 8.88835449e-01, -4.58226522e-01, 8.32998052e+00],
[ -5.40640817e-01, -8.41253533e-01, 9.99597663e+00],
[ -1.00000000e+00, -4.28626380e-16, 1.09955743e+01]]))