Updating GazeAnalysis test.

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__':