diff options
Diffstat (limited to 'src/argaze.test')
| -rw-r--r-- | src/argaze.test/ArFeatures.py | 6 | ||||
| -rw-r--r-- | src/argaze.test/utils/environment.json | 17 |
2 files changed, 19 insertions, 4 deletions
diff --git a/src/argaze.test/ArFeatures.py b/src/argaze.test/ArFeatures.py index af8cfa8..054416a 100644 --- a/src/argaze.test/ArFeatures.py +++ b/src/argaze.test/ArFeatures.py @@ -37,11 +37,11 @@ class TestArEnvironmentClass(unittest.TestCase): self.assertIsNone(numpy.testing.assert_array_equal(ar_environment.aruco_detector.camera.D, [-1.0, -0.5, 0.0, 0.5, 1.0])) # Check environment scenes - self.assertEqual(len(ar_environment.scenes), 1) - self.assertIsNone(numpy.testing.assert_array_equal(list(ar_environment.scenes.keys()), ["TestScene"])) + self.assertEqual(len(ar_environment.scenes), 2) + self.assertIsNone(numpy.testing.assert_array_equal(list(ar_environment.scenes.keys()), ["TestSceneA", "TestSceneB"])) # Load test scene - ar_scene = ar_environment.scenes["TestScene"] + ar_scene = ar_environment.scenes["TestSceneA"] # Check Aruco scene self.assertEqual(len(ar_scene.aruco_scene.places), 2) diff --git a/src/argaze.test/utils/environment.json b/src/argaze.test/utils/environment.json index e4cf43a..9476a0d 100644 --- a/src/argaze.test/utils/environment.json +++ b/src/argaze.test/utils/environment.json @@ -40,7 +40,22 @@ "aprilTagDeglitch": 1 } }, - "TestScene" : { + "TestSceneA" : { + "aruco_scene": { + "0": { + "translation": [1, 0, 0], + "rotation": [0, 0, 0] + }, + "1": { + "translation": [0, 1, 0], + "rotation": [0, 90, 0] + } + }, + "aoi_scene": "aoi.obj", + "angle_tolerance": 1.0, + "distance_tolerance": 2.0 + }, + "TestSceneB" : { "aruco_scene": { "0": { "translation": [1, 0, 0], |