diff options
Diffstat (limited to 'src/argaze.test/ArUcoMarkers')
| -rw-r--r-- | src/argaze.test/ArUcoMarkers/ArUcoScene.py | 10 |
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.""" |