From 8585f3a1550e6c2defc73e29ce496e307993f08f Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Thu, 6 Jul 2023 18:13:49 +0200
Subject: Test finish method.

---
 src/argaze.test/GazeFeatures.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'src/argaze.test/GazeFeatures.py')

diff --git a/src/argaze.test/GazeFeatures.py b/src/argaze.test/GazeFeatures.py
index b8d173c..85ba362 100644
--- a/src/argaze.test/GazeFeatures.py
+++ b/src/argaze.test/GazeFeatures.py
@@ -302,6 +302,24 @@ class TestGazeMovementClass(unittest.TestCase):
         self.assertEqual(abstract_gaze_movement.valid, False)
         self.assertEqual(abstract_gaze_movement.finished, False)
 
+    def test_finish(self):
+        """Test GazeMovement finishing."""
+        
+        abstract_gaze_movement = GazeFeatures.GazeMovement(random_gaze_positions(2))
+
+        abstract_gaze_movement_ref = abstract_gaze_movement
+
+        # Check abstract GazeMovement and its reference
+        self.assertEqual(abstract_gaze_movement.finished, False)
+        self.assertEqual(abstract_gaze_movement_ref.finished, False)
+
+        # Set gaze movement as finished
+        abstract_gaze_movement.finish()
+
+        # Check abstract GazeMovement and its reference
+        self.assertEqual(abstract_gaze_movement.finished, True)
+        self.assertEqual(abstract_gaze_movement_ref.finished, True)
+
 class TestUnvalidGazeMovementClass(unittest.TestCase):
     """Test UnvalidGazeMovement class."""
 
-- 
cgit v1.1