From e35e90d161ffb9202459631a0049448cde905b3c Mon Sep 17 00:00:00 2001 From: Théo de la Hogue Date: Wed, 31 May 2023 15:35:28 +0200 Subject: Testing all gaze analysis algorithm. Unifying paper citation. --- .../GazeAnalysis/NearestNeighborIndex.py | 40 ++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 src/argaze.test/GazeAnalysis/NearestNeighborIndex.py (limited to 'src/argaze.test/GazeAnalysis/NearestNeighborIndex.py') 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 -- cgit v1.1