Testing ArUcoMarkers submodule.

1 files changed, 27 insertions, 0 deletions

diff --git a/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py b/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py
new file mode 100644
index 0000000..58849a6
--- /dev/null
+++ b/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py
@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+
+import unittest
+
+from argaze.ArUcoMarkers import ArUcoMarkersDictionary
+
+class TestArUcoMarkersDictionaryClass(unittest.TestCase):
+    """Test ArUcoMarkersDictionary class."""
+
+    def test_new(self):
+        """Test ArUcoMarkersDictionary creation."""
+
+        # Check that ArUcoMarkersDictionary creation fails with bad name
+        with self.assertRaises(NameError):
+
+            aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('BAD_DICT_NAME')
+
+        # Check DICT_ARUCO_ORIGINAL ArUcoMarkersDictionary creation
+        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('DICT_ARUCO_ORIGINAL')
+
+        self.assertEqual(aruco_dictionary.name, 'DICT_ARUCO_ORIGINAL')
+        self.assertEqual(aruco_dictionary.format, '5X5')
+        self.assertEqual(aruco_dictionary.number, 1024)
+
+if __name__ == '__main__':
+
+    unittest.main() 
\ No newline at end of file