aboutsummaryrefslogtreecommitdiff
path: root/src/argaze.test/ArUcoMarkers/ArUcoScene.py
diff options
context:
space:
mode:
authorThéo de la Hogue2023-05-30 13:34:44 +0200
committerThéo de la Hogue2023-05-30 13:34:44 +0200
commit2c296a5adc88d9fde36d6c8e24ca275a7088a8f9 (patch)
treeda87a530eb8b64355df90377d6abf06d6df07763 /src/argaze.test/ArUcoMarkers/ArUcoScene.py
parentb1c5eaa9974d91263083ea457b65ad60c20cca85 (diff)
downloadargaze-2c296a5adc88d9fde36d6c8e24ca275a7088a8f9.zip
argaze-2c296a5adc88d9fde36d6c8e24ca275a7088a8f9.tar.gz
argaze-2c296a5adc88d9fde36d6c8e24ca275a7088a8f9.tar.bz2
argaze-2c296a5adc88d9fde36d6c8e24ca275a7088a8f9.tar.xz
Updating tests.
Diffstat (limited to 'src/argaze.test/ArUcoMarkers/ArUcoScene.py')
-rw-r--r--src/argaze.test/ArUcoMarkers/ArUcoScene.py10
1 files changed, 7 insertions, 3 deletions
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."""