diff options
| -rw-r--r-- | src/argaze.test/ArScene.py | 10 |
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 |