From 7a4692429fa9bcd97e92483a24f335db84add819 Mon Sep 17 00:00:00 2001 From: Romain Martinez Date: Wed, 3 Jun 2020 16:29:56 -0400 Subject: [PATCH] correct from_averaged_rototrans test --- tests/test_processing_rt.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/tests/test_processing_rt.py b/tests/test_processing_rt.py index 22df3e7..a61b211 100644 --- a/tests/test_processing_rt.py +++ b/tests/test_processing_rt.py @@ -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 + )