diff options
Diffstat (limited to 'src/argaze.test/GazeAnalysis')
-rw-r--r-- | src/argaze.test/GazeAnalysis/Entropy.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/src/argaze.test/GazeAnalysis/Entropy.py b/src/argaze.test/GazeAnalysis/Entropy.py new file mode 100644 index 0000000..47d5556 --- /dev/null +++ b/src/argaze.test/GazeAnalysis/Entropy.py @@ -0,0 +1,41 @@ +#!/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 Entropy, 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', 'Shu', 'Foo', 'Bar', 'Shu', 'Foo']) + + entropy_analyzer = Entropy.AOIScanPathAnalyzer() + transition_matrix_analyser = TransitionMatrix.AOIScanPathAnalyzer() + + transition_matrix_probabilities, transition_matrix_density = transition_matrix_analyser.analyze(aoi_scan_path) + stationary_entropy, transition_entropy = entropy_analyzer.analyze(aoi_scan_path, transition_matrix_probabilities) + + # Check aoi scan path + self.assertEqual(len(aoi_scan_path), 9) + + # Check entropy analysis + self.assertAlmostEqual(stationary_entropy, 1.09, 1) + self.assertAlmostEqual(transition_entropy, 0, 1) + +if __name__ == '__main__': + + unittest.main()
\ No newline at end of file |