From 4d0de7c804914a55977635ec6bc46beb0cf7808a Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Wed, 17 Apr 2024 13:32:51 +0200 Subject: Renaming ArUcoMarkers into ArUcoMarker --- .../ArUcoMarker/ArUcoMarkerDictionary.py | 50 ++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 src/argaze.test/ArUcoMarker/ArUcoMarkerDictionary.py (limited to 'src/argaze.test/ArUcoMarker/ArUcoMarkerDictionary.py') diff --git a/src/argaze.test/ArUcoMarker/ArUcoMarkerDictionary.py b/src/argaze.test/ArUcoMarker/ArUcoMarkerDictionary.py new file mode 100644 index 0000000..2f80a67 --- /dev/null +++ b/src/argaze.test/ArUcoMarker/ArUcoMarkerDictionary.py @@ -0,0 +1,50 @@ +""" + +This program is free software: you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free Software +Foundation, either version 3 of the License, or (at your option) any later +version. +This program is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. +You should have received a copy of the GNU General Public License along with +this program. If not, see . +""" + +__author__ = "Théo de la Hogue" +__credits__ = [] +__copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)" +__license__ = "GPLv3" + +import unittest + +from argaze.ArUcoMarker import ArUcoMarkerDictionary + +class TestArUcoMarkerDictionaryClass(unittest.TestCase): + """Test ArUcoMarkerDictionary class.""" + + def test_new(self): + """Test ArUcoMarkerDictionary creation.""" + + # Check that ArUcoMarkerDictionary creation fails with bad name + with self.assertRaises(NameError): + + aruco_dictionary = ArUcoMarkerDictionary.ArUcoMarkerDictionary('BAD_DICT_NAME') + + # Check default ArUcoMarkerDictionary creation + aruco_dictionary = ArUcoMarkerDictionary.ArUcoMarkerDictionary() + + self.assertEqual(aruco_dictionary.name, 'DICT_ARUCO_ORIGINAL') + self.assertEqual(aruco_dictionary.format, '5X5') + self.assertEqual(aruco_dictionary.number, 1024) + + # Check DICT_APRILTAG_16h5 ArUcoMarkerDictionary creation + aruco_dictionary = ArUcoMarkerDictionary.ArUcoMarkerDictionary('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 -- cgit v1.1