Default ArUco Dictionary creation.

1 files changed, 9 insertions, 2 deletions

diff --git a/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py b/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py
index 58849a6..e27580f 100644
--- a/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py
+++ b/src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py
@@ -15,13 +15,20 @@ class TestArUcoMarkersDictionaryClass(unittest.TestCase):
 
             aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('BAD_DICT_NAME')
 
-        # Check DICT_ARUCO_ORIGINAL ArUcoMarkersDictionary creation
-        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('DICT_ARUCO_ORIGINAL')
+        # Check default ArUcoMarkersDictionary creation
+        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary()
 
         self.assertEqual(aruco_dictionary.name, 'DICT_ARUCO_ORIGINAL')
         self.assertEqual(aruco_dictionary.format, '5X5')
         self.assertEqual(aruco_dictionary.number, 1024)
 
+        # Check DICT_APRILTAG_16h5 ArUcoMarkersDictionary creation
+        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('DICT_APRILTAG_16h5')
+
+        self.assertEqual(aruco_dictionary.name, 'DICT_APRILTAG_16h5')
+        self.assertEqual(aruco_dictionary.format, '4X4')
+        self.assertEqual(aruco_dictionary.number, 30)
+
 if __name__ == '__main__':
 
     unittest.main() 
\ No newline at end of file