diff options
Diffstat (limited to 'src/argaze.test/GazeAnalysis/Entropy.py')
-rw-r--r-- | src/argaze.test/GazeAnalysis/Entropy.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/argaze.test/GazeAnalysis/Entropy.py b/src/argaze.test/GazeAnalysis/Entropy.py index b69f329..f84eca0 100644 --- a/src/argaze.test/GazeAnalysis/Entropy.py +++ b/src/argaze.test/GazeAnalysis/Entropy.py @@ -21,20 +21,20 @@ class TestAOIScanPathAnalyzer(unittest.TestCase): def test_analyze(self): """Test analyze method.""" - entropy_analyzer = Entropy.AOIScanPathAnalyzer() transition_matrix_analyser = TransitionMatrix.AOIScanPathAnalyzer() + entropy_analyzer = Entropy.AOIScanPathAnalyzer(transition_matrix_analyser) aoi_scan_path = GazeFeaturesTest.build_aoi_scan_path(['Foo', 'Bar', 'Shu'], ['Bar', 'Shu', 'Foo', 'Bar', 'Shu', 'Foo', 'Bar', 'Shu', 'Foo']) # Check aoi scan path self.assertEqual(len(aoi_scan_path), 9) - 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) + transition_matrix_analyser.analyze(aoi_scan_path) + entropy_analyzer.analyze(aoi_scan_path) # Check entropy analysis - self.assertAlmostEqual(stationary_entropy, 1.09, 1) - self.assertAlmostEqual(transition_entropy, 0, 1) + self.assertAlmostEqual(entropy_analyzer.stationary_entropy, 1.09, 1) + self.assertAlmostEqual(entropy_analyzer.transition_entropy, 0, 1) if __name__ == '__main__': |