From 0a7154cbaa43bb14aae62fc4324ad24c60949472 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Tue, 29 Nov 2022 19:08:14 +0100
Subject: Renaming ArGazeProject into ArGazeScene. Removing useless old
 ArUcoSet related class.

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

(limited to 'src/argaze.test/ArUcoMarkers/ArUcoTracker.py')

diff --git a/src/argaze.test/ArUcoMarkers/ArUcoTracker.py b/src/argaze.test/ArUcoMarkers/ArUcoTracker.py
index acc0000..87373ea 100644
--- a/src/argaze.test/ArUcoMarkers/ArUcoTracker.py
+++ b/src/argaze.test/ArUcoMarkers/ArUcoTracker.py
@@ -9,6 +9,29 @@ from argaze.ArUcoMarkers import ArUcoMarkersDictionary, ArUcoCamera, ArUcoTracke
 import cv2 as cv
 import numpy
 
+class TestTrackingDataClass(unittest.TestCase):
+    """Test TrackingData class."""
+
+    def test_from_json(self):
+        """Test TrackingData creation from json file."""
+        
+        # Edit traking data file path
+        current_directory = os.path.dirname(os.path.abspath(__file__))
+        json_filepath = os.path.join(current_directory, 'utils/tracker.json')
+
+        # Load project
+        tracking_data = ArUcoTracker.TrackingData.from_json(json_filepath)
+
+        # Check data
+        self.assertEqual(tracking_data.cornerRefinementMethod, 3)
+        self.assertEqual(tracking_data.aprilTagQuadSigma, 2)
+        self.assertEqual(tracking_data.aprilTagDeglitch, 1)
+
+        # Check bad data access fails
+        with self.assertRaises(AttributeError):
+
+            tracking_data.unknown_data = 1
+
 class TestArUcoTrackerClass(unittest.TestCase):
     """Test ArUcoTracker class."""
 
-- 
cgit v1.1