Skip to content

Commit

Permalink
correct from_averaged_rototrans test
Browse files Browse the repository at this point in the history
  • Loading branch information
romainmartinez committed Jun 3, 2020
1 parent f13846b commit 7a46924
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions tests/test_processing_rt.py
Expand Up @@ -241,13 +241,19 @@ def test_rt_transpose():


def test_average_rt():
angles = Angles(np.random.rand(3, 1, 100))
seq = "xyz"
angles_size = (3, 1, 100)
ref_angles_from_rt_mean = [0.25265133, 0.57436872, 0.79133042]

restard_seed()
angles = Angles.from_random_data(size=angles_size)
# min-max normalization to keep the angles low
angles = angles.pipe(lambda x: (x - x.min()) / (x.max() - x.min()))

rt = Rototrans.from_euler_angles(angles, seq)
rt_mean = Rototrans.from_averaged_rototrans(rt)
angles_mean = Angles.from_rototrans(rt_mean, seq).isel(time=0)

angles_mean_ref = Angles.from_rototrans(rt, seq).mean(dim="time")
angles_from_rt_mean = Angles.from_rototrans(rt_mean, seq)

np.testing.assert_array_almost_equal(angles_mean, angles_mean_ref, decimal=2)
np.testing.assert_almost_equal(
angles_from_rt_mean.data.ravel(), ref_angles_from_rt_mean, decimal=4
)

0 comments on commit 7a46924

Please sign in to comment.