fix assert on angles

release/4.3a0
John Lambert 2021-07-13 00:16:24 -06:00 committed by GitHub
parent d7151ed284
commit c4a4e13196
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions

View File

@ -196,11 +196,13 @@ class TestShonanAveraging(GtsamTestCase):
wRi_list = [result_values.atRot2(i) for i in range(num_images)] wRi_list = [result_values.atRot2(i) for i in range(num_images)]
thetas_deg = np.array([wRi.degrees() for wRi in wRi_list]) thetas_deg = np.array([wRi.degrees() for wRi in wRi_list])
thetas_deg -= max(thetas_deg)
# map all angles to [0,360) # map all angles to [0,360)
thetas_deg = thetas_deg % 360 thetas_deg = thetas_deg % 360
thetas_deg -= max(thetas_deg)
expected_thetas_deg = np.array([0.0, 90.0, 0.0]) expected_thetas_deg = np.array([0.0, 90.0, 0.0])
np.testing.assert_allclose(thetas_deg, expected_thetas_deg) np.testing.assert_allclose(thetas_deg, expected_thetas_deg, atol=0.1)
if __name__ == "__main__": if __name__ == "__main__":