diff options
Diffstat (limited to 'src/argaze.test/ArGazeProject.py')
-rw-r--r-- | src/argaze.test/ArGazeProject.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/argaze.test/ArGazeProject.py b/src/argaze.test/ArGazeProject.py index f37e1f5..8e449bf 100644 --- a/src/argaze.test/ArGazeProject.py +++ b/src/argaze.test/ArGazeProject.py @@ -5,6 +5,7 @@ import os from argaze import ArGazeProject +import numpy class TestArGazeProjectClass(unittest.TestCase): """Test ArGazeProject class.""" @@ -23,8 +24,10 @@ class TestArGazeProjectClass(unittest.TestCase): self.assertEqual(argaze_project.name, "TestProject") # Check ArUco camera - print(argaze_project.aruco_camera) - self.assertEqual(argaze_project.aruco_camera.rms, 0.) + self.assertEqual(argaze_project.aruco_camera.rms, 1.0) + self.assertIsNone(numpy.testing.assert_array_equal(argaze_project.aruco_camera.dimensions, [1920, 1080])) + self.assertIsNone(numpy.testing.assert_array_equal(argaze_project.aruco_camera.K, [[1.0, 0.0, 1.0], [0.0, 1.0, 1.0], [0.0, 0.0, 1.0]])) + self.assertIsNone(numpy.testing.assert_array_equal(argaze_project.aruco_camera.D, [-1.0, -0.5, 0.0, 0.5, 1.0])) if __name__ == '__main__': |