From 2c296a5adc88d9fde36d6c8e24ca275a7088a8f9 Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Tue, 30 May 2023 13:34:44 +0200 Subject: Updating tests. --- src/argaze.test/ArUcoMarkers/ArUcoScene.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'src/argaze.test/ArUcoMarkers/ArUcoScene.py') diff --git a/src/argaze.test/ArUcoMarkers/ArUcoScene.py b/src/argaze.test/ArUcoMarkers/ArUcoScene.py index ee618d8..5e24995 100644 --- a/src/argaze.test/ArUcoMarkers/ArUcoScene.py +++ b/src/argaze.test/ArUcoMarkers/ArUcoScene.py @@ -130,7 +130,8 @@ class TestArUcoSceneClass(unittest.TestCase): # Check consistent markers, unconsistent markers and unconsistencies self.assertEqual(len(consistent_markers), 3) self.assertEqual(len(unconsistent_markers), 0) - self.assertEqual(len(unconsistencies), 0) + self.assertEqual(len(unconsistencies['rotation']), 0) + self.assertEqual(len(unconsistencies['translation']), 0) self.assertIsNone(numpy.testing.assert_array_equal(list(consistent_markers.keys()), self.aruco_scene.identifiers)) @@ -143,10 +144,13 @@ class TestArUcoSceneClass(unittest.TestCase): # Check consistent markers, unconsistent markers and unconsistencies self.assertEqual(len(consistent_markers), 2) self.assertEqual(len(unconsistent_markers), 1) - self.assertEqual(len(unconsistencies), 2) + self.assertEqual(len(unconsistencies['rotation']), 0) + self.assertEqual(len(unconsistencies['translation']), 2) self.assertIsNone(numpy.testing.assert_array_equal(list(unconsistent_markers.keys()), [2])) - self.assertIsNone(numpy.testing.assert_array_equal(list(unconsistencies.keys()), ['0/2 distance', '1/2 distance'])) + self.assertIsNone(numpy.testing.assert_array_equal(list(unconsistencies['translation'].keys()), ['0/2', '1/2'])) + self.assertIsNone(numpy.testing.assert_array_equal(list(unconsistencies['translation']['0/2'].keys()), ['current', 'expected'])) + self.assertIsNone(numpy.testing.assert_array_equal(list(unconsistencies['translation']['1/2'].keys()), ['current', 'expected'])) def test_estimate_pose_from_single_marker(self): """Test ArUcoScene pose estimation from single marker.""" -- cgit v1.1