aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/argaze.test/ArScene.py10
1 files changed, 6 insertions, 4 deletions
diff --git a/src/argaze.test/ArScene.py b/src/argaze.test/ArScene.py
index cfb949b..82154f3 100644
--- a/src/argaze.test/ArScene.py
+++ b/src/argaze.test/ArScene.py
@@ -37,17 +37,19 @@ class TestArSceneClass(unittest.TestCase):
self.assertEqual(ar_scene.aruco_tracker.tracking_data.aprilTagDeglitch, 1)
# Check ArUco scene
- self.assertEqual(ar_scene.aruco_scene.angle_tolerance, 1.0)
- self.assertEqual(ar_scene.aruco_scene.distance_tolerance, 2.0)
+ self.assertEqual(ar_scene.angle_tolerance, 1.0)
+ self.assertEqual(ar_scene.distance_tolerance, 2.0)
self.assertEqual(len(ar_scene.aruco_scene.places), 2)
+ print(ar_scene.aruco_scene.places['A'].rotation)
+
# Check ArUco scene places
self.assertIsNone(numpy.testing.assert_array_equal(ar_scene.aruco_scene.places['A'].translation, [1, 0, 0]))
- self.assertIsNone(numpy.testing.assert_array_equal(ar_scene.aruco_scene.places['A'].rotation, [0, 0, 0]))
+ self.assertIsNone(numpy.testing.assert_array_equal(ar_scene.aruco_scene.places['A'].rotation, [[1.,0.,0.],[0.,1.,0.],[0.,0.,1.]]))
self.assertEqual(ar_scene.aruco_scene.places['A'].marker.identifier, 0)
self.assertIsNone(numpy.testing.assert_array_equal(ar_scene.aruco_scene.places['B'].translation, [0, 1, 0]))
- self.assertIsNone(numpy.testing.assert_array_equal(ar_scene.aruco_scene.places['B'].rotation, [0, 90, 0]))
+ self.assertIsNone(numpy.testing.assert_array_almost_equal(ar_scene.aruco_scene.places['B'].rotation, [[0.,0.,1.],[0., 1.,0.],[-1.,0.,0.]]))
self.assertEqual(ar_scene.aruco_scene.places['B'].marker.identifier, 1)
# Check AOI scene