diff options
Diffstat (limited to 'src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py')
-rw-r--r-- | src/argaze.test/ArUcoMarkers/ArUcoMarkersDictionary.py | 27 |
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 |