Testing all gaze analysis algorithm. Unifying paper citation.

1 files changed, 40 insertions, 0 deletions

diff --git a/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py b/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
new file mode 100644
index 0000000..fb7d4ec
--- /dev/null
+++ b/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
@@ -0,0 +1,40 @@
+#!/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 NearestNeighborIndex
+from argaze.utils import MiscFeatures
+
+GazeFeaturesTest = MiscFeatures.importFromTestPackage('GazeFeatures')
+
+class TestScanPathAnalyzer(unittest.TestCase):
+    """Test ScanPathAnalyzer class."""
+
+    def test_analyze(self):
+        """Test analyze."""
+
+        nni_analyzer = NearestNeighborIndex.ScanPathAnalyzer()
+
+        screen_dimension = (100, 100)
+        scan_path = GazeFeaturesTest.build_scan_path(6, screen_dimension)
+
+        # Check aoi scan path
+        self.assertEqual(len(scan_path), 6)
+
+        nni = nni_analyzer.analyze(scan_path, screen_dimension)
+
+        # Check NNI
+        self.assertGreaterEqual(nni, 0)
+        self.assertLessEqual(nni, 1)
+
+if __name__ == '__main__':
+
+    unittest.main()
\ No newline at end of file