diff options
Diffstat (limited to 'src/argaze.test/GazeAnalysis/KCoefficient.py')
-rw-r--r-- | src/argaze.test/GazeAnalysis/KCoefficient.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/src/argaze.test/GazeAnalysis/KCoefficient.py b/src/argaze.test/GazeAnalysis/KCoefficient.py new file mode 100644 index 0000000..07dff79 --- /dev/null +++ b/src/argaze.test/GazeAnalysis/KCoefficient.py @@ -0,0 +1,56 @@ +#!/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 KCoefficient +from argaze.utils import MiscFeatures + +GazeFeaturesTest = MiscFeatures.importFromTestPackage('GazeFeatures') + +class TestScanPathAnalyzer(unittest.TestCase): + """Test ScanPathAnalyzer class.""" + + def test_analyze(self): + """Test analyze method.""" + + kcoeff_analyzer = KCoefficient.AOIScanPathAnalyzer() + + scan_path = GazeFeaturesTest.build_scan_path(10) + + # Check scan path + self.assertEqual(len(scan_path), 10) + + K = kcoeff_analyzer.analyze(scan_path) + + # Check that K coefficient is almost equal to 0 + self.assertAlmostEqual(K, 0) + +class TestAOIScanPathAnalyzer(unittest.TestCase): + """Test AOIScanPathAnalyzer class.""" + + def test_analyze(self): + """Test analyze method.""" + + kcoeff_analyzer = KCoefficient.AOIScanPathAnalyzer() + + aoi_scan_path = GazeFeaturesTest.build_aoi_scan_path(['Foo', 'Bar', 'Shu'], ['Bar', 'Shu', 'Foo', 'Bar']) + + # Check aoi scan path + self.assertEqual(len(aoi_scan_path), 4) + + K = kcoeff_analyzer.analyze(aoi_scan_path) + + # Check that K coefficient is almost equal to 0 + self.assertAlmostEqual(K, 0) + +if __name__ == '__main__': + + unittest.main()
\ No newline at end of file |