1 files changed, 23 insertions, 0 deletions
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."""