From 516ed19d801f9fe27ad3b70e72e3b0f9c057efdf Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Wed, 31 May 2023 11:57:32 +0200
Subject: Testing and commenting transition matrix.

---
 src/argaze.test/GazeAnalysis/TransitionMatrix.py | 51 ++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 src/argaze.test/GazeAnalysis/TransitionMatrix.py

(limited to 'src/argaze.test/GazeAnalysis/TransitionMatrix.py')

diff --git a/src/argaze.test/GazeAnalysis/TransitionMatrix.py b/src/argaze.test/GazeAnalysis/TransitionMatrix.py
new file mode 100644
index 0000000..14a34ce
--- /dev/null
+++ b/src/argaze.test/GazeAnalysis/TransitionMatrix.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+
+""" """
+
+__author__ = "Théo de la Hogue"
+__credits__ = []
+__copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)"
+__license__ = "BSD"
+
+import unittest
+
+from argaze import GazeFeatures
+from argaze.GazeAnalysis import TransitionMatrix
+from argaze.utils import MiscFeatures
+
+GazeFeaturesTest = MiscFeatures.importFromTestPackage('GazeFeatures')
+
+class TestAOIScanPathAnalyzer(unittest.TestCase):
+    """Test AOIScanPathAnalyzer class."""
+
+    def test_analyse(self):
+        """Test analyse method."""
+
+        aoi_scan_path = GazeFeaturesTest.build_aoi_scan_path(['Foo', 'Bar', 'Shu'], ['Bar', 'Shu', 'Foo', 'Bar'])
+
+        transition_matrix_analyser = TransitionMatrix.AOIScanPathAnalyzer()
+        transition_matrix_probabilities, transition_matrix_density = transition_matrix_analyser.analyze(aoi_scan_path)
+
+        # Check aoi scan path
+        self.assertEqual(len(aoi_scan_path), 4)
+
+        # Check transition matrix probabilities ([destination][departure])
+        self.assertEqual(transition_matrix_probabilities['Foo']['Foo'], 0)
+        self.assertEqual(transition_matrix_probabilities['Bar']['Bar'], 0)
+        self.assertEqual(transition_matrix_probabilities['Shu']['Shu'], 0)
+
+        self.assertEqual(transition_matrix_probabilities['Foo']['Bar'], 0)
+        self.assertEqual(transition_matrix_probabilities['Foo']['Shu'], 1)
+
+        self.assertEqual(transition_matrix_probabilities['Bar']['Foo'], 1)
+        self.assertEqual(transition_matrix_probabilities['Bar']['Shu'], 0)
+
+        self.assertEqual(transition_matrix_probabilities['Shu']['Foo'], 0)
+        self.assertEqual(transition_matrix_probabilities['Shu']['Bar'], 1)
+
+        # Check transition matrix density
+        self.assertEqual(transition_matrix_density, 0.3333333333333333)
+
+if __name__ == '__main__':
+
+    unittest.main()
\ No newline at end of file
-- 
cgit v1.1