From 6bb8f307f67275543ec50b51fc98a19483b0c066 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 28 Mar 2023 14:35:29 +0200
Subject: Making ArUcoDetector as dataclass.

---
 src/argaze.test/ArUcoMarkers/ArUcoDetector.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/argaze.test')

diff --git a/src/argaze.test/ArUcoMarkers/ArUcoDetector.py b/src/argaze.test/ArUcoMarkers/ArUcoDetector.py
index 3272f1c..5a45fad 100644
--- a/src/argaze.test/ArUcoMarkers/ArUcoDetector.py
+++ b/src/argaze.test/ArUcoMarkers/ArUcoDetector.py
@@ -42,6 +42,17 @@ class TestArUcoDetectorClass(unittest.TestCase):
 
          # Check ArUcoDetector creation
         self.assertEqual(aruco_detector.dictionary.name, 'DICT_ARUCO_ORIGINAL')
+        self.assertEqual(aruco_detector.marker_size, 3)
+        self.assertIsNone(numpy.testing.assert_array_equal(aruco_detector.camera.dimensions, [0, 0]))
+        self.assertEqual(aruco_detector.detected_markers_number, 0)
+        self.assertEqual(aruco_detector.detected_markers, {})
+
+        aruco_dictionary = ArUcoMarkersDictionary.ArUcoMarkersDictionary('DICT_APRILTAG_16h5')
+        aruco_detector = ArUcoDetector.ArUcoDetector(aruco_dictionary, 5.2)
+
+         # Check ArUcoDetector creation
+        self.assertEqual(aruco_detector.dictionary.name, 'DICT_APRILTAG_16h5')
+        self.assertEqual(aruco_detector.marker_size, 5.2)
         self.assertIsNone(numpy.testing.assert_array_equal(aruco_detector.camera.dimensions, [0, 0]))
         self.assertEqual(aruco_detector.detected_markers_number, 0)
         self.assertEqual(aruco_detector.detected_markers, {})
@@ -58,6 +69,7 @@ class TestArUcoDetectorClass(unittest.TestCase):
 
          # Check ArUcoDetector creation
         self.assertEqual(aruco_detector.dictionary.name, 'DICT_ARUCO_ORIGINAL')
+        self.assertEqual(aruco_detector.marker_size, 3)
         self.assertIsNone(numpy.testing.assert_array_equal(aruco_detector.camera.dimensions, [1920, 1080]))
         self.assertEqual(aruco_detector.parameters.cornerRefinementMethod, 3)
         self.assertEqual(aruco_detector.parameters.aprilTagQuadSigma, 2)
-- 
cgit v1.1