Testing all gaze analysis algorithm. Unifying paper citation.

1 files changed, 48 insertions, 24 deletions

diff --git a/src/argaze.test/GazeFeatures.py b/src/argaze.test/GazeFeatures.py
index 8206baf..f73eefe 100644
--- a/src/argaze.test/GazeFeatures.py
+++ b/src/argaze.test/GazeFeatures.py
@@ -8,12 +8,13 @@ __copyright__ = "Copyright 2023, Ecole Nationale de l'Aviation Civile (ENAC)"
 __license__ = "BSD"
 
 import unittest
+from dataclasses import dataclass
 
 from argaze import GazeFeatures
 
 import numpy
 
-def random_gaze_positions(size):
+def random_gaze_positions(size, screen_dimension: tuple[float, float] = (1, 1)):
     """ Generate random TimeStampedGazePsoitions for testing purpose.
         Timestamps are current time.
         GazePositions are random values.
@@ -27,13 +28,36 @@ def random_gaze_positions(size):
     for i in range(0, size):
 
         # Edit gaze position
-        random_gaze_position = GazeFeatures.GazePosition((random.random(), random.random()))
+        random_gaze_position = GazeFeatures.GazePosition((random.random() * screen_dimension[0], random.random() * screen_dimension[1]))
 
         # Store gaze position
         ts_gaze_positions[time.time()] = random_gaze_position
 
     return ts_gaze_positions
 
+@dataclass(frozen=True)
+class TestFixation(GazeFeatures.Fixation):
+    """Define basic fixation class for test."""
+
+    def __post_init__(self):
+
+        super().__post_init__()
+
+        points = self.positions.values()
+        points_x, points_y = [p[0] for p in points], [p[1] for p in points]
+        points_array = numpy.column_stack([points_x, points_y])
+        centroid_array = numpy.array([numpy.mean(points_x), numpy.mean(points_y)])
+
+        # Update frozen focus attribute using centroid
+        object.__setattr__(self, 'focus', (centroid_array[0], centroid_array[1]))
+
+@dataclass(frozen=True)
+class TestSaccade(GazeFeatures.Saccade):
+    """Define basic saccade for test."""
+
+    def __post_init__(self):
+        super().__post_init__()
+
 class TestGazePositionClass(unittest.TestCase):
     """Test GazePosition class."""
 
@@ -222,8 +246,8 @@ class TestScanStepClass(unittest.TestCase):
     def test_new(self):
         """Test ScanStep creation."""
 
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        fixation = TestFixation(random_gaze_positions(10))
+        saccade = TestSaccade(random_gaze_positions(2))
 
         scan_step = GazeFeatures.ScanStep(fixation, saccade)
 
@@ -232,18 +256,18 @@ class TestScanStepClass(unittest.TestCase):
         self.assertEqual(scan_step.last_saccade, saccade)
         self.assertGreater(scan_step.duration, 0)
 
-def build_scan_path(size):
+def build_scan_path(size, screen_dimension: tuple[float, float] = (1, 1)):
     """Build scan path"""
 
     scan_path = GazeFeatures.ScanPath()
 
     for i in range(size):
 
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10, screen_dimension))
         ts, _ = fixation.positions.first
         scan_path.append_fixation(ts, fixation)
 
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2, screen_dimension))
         ts, _ = saccade.positions.first
         scan_path.append_saccade(ts, saccade)
 
@@ -266,7 +290,7 @@ class TestScanPathClass(unittest.TestCase):
         scan_path = GazeFeatures.ScanPath()
 
         # Append a saccade that should be ignored
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade.positions.first
 
         new_step = scan_path.append_saccade(ts, saccade)
@@ -276,7 +300,7 @@ class TestScanPathClass(unittest.TestCase):
         self.assertEqual(new_step, None)
 
         # Append first fixation
-        fixation_A = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation_A = TestFixation(random_gaze_positions(10))
         ts, _ = fixation_A.positions.first
 
         new_step = scan_path.append_fixation(ts, fixation_A)
@@ -286,7 +310,7 @@ class TestScanPathClass(unittest.TestCase):
         self.assertEqual(new_step, None)
 
         # Append consecutive saccade
-        saccade_A = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade_A = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade_A.positions.first
 
         new_step = scan_path.append_saccade(ts, saccade_A)
@@ -297,7 +321,7 @@ class TestScanPathClass(unittest.TestCase):
         self.assertEqual(new_step.last_saccade, saccade_A)
 
         # Append 2 consecutive fixations then a saccade
-        fixation_B1 = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation_B1 = TestFixation(random_gaze_positions(10))
         ts, _ = fixation_B1.positions.first
 
         new_step = scan_path.append_fixation(ts, fixation_B1)
@@ -306,7 +330,7 @@ class TestScanPathClass(unittest.TestCase):
         self.assertEqual(len(scan_path), 1)
         self.assertEqual(new_step, None)
 
-        fixation_B2 = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation_B2 = TestFixation(random_gaze_positions(10))
         ts, _ = fixation_B2.positions.first
 
         new_step = scan_path.append_fixation(ts, fixation_B2)
@@ -315,7 +339,7 @@ class TestScanPathClass(unittest.TestCase):
         self.assertEqual(len(scan_path), 1)
         self.assertEqual(new_step, None)
 
-        saccade_B = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade_B = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade_B.positions.first
 
         new_step = scan_path.append_saccade(ts, saccade_B)
@@ -333,11 +357,11 @@ class TestAOIScanStepClass(unittest.TestCase):
 
         movements = GazeFeatures.TimeStampedGazeMovements()
 
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
         movements[ts] = fixation
 
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade.positions.first
         movements[ts] = saccade
 
@@ -355,11 +379,11 @@ class TestAOIScanStepClass(unittest.TestCase):
 
         movements = GazeFeatures.TimeStampedGazeMovements()
 
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade.positions.first
         movements[ts] = saccade
 
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
         movements[ts] = fixation
 
@@ -378,11 +402,11 @@ def build_aoi_scan_path(expected_aois, aoi_path):
 
     for aoi in aoi_path:
 
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
         aoi_scan_path.append_fixation(ts, fixation, aoi)
 
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade.positions.first
         aoi_scan_path.append_saccade(ts, saccade)
 
@@ -405,7 +429,7 @@ class TestAOIScanPathClass(unittest.TestCase):
         aoi_scan_path = GazeFeatures.AOIScanPath(['Foo', 'Bar'])
 
         # Append fixation on A aoi
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
 
         new_step = aoi_scan_path.append_fixation(ts, fixation, 'Foo')
@@ -415,7 +439,7 @@ class TestAOIScanPathClass(unittest.TestCase):
         self.assertEqual(new_step, None)
 
         # Append saccade
-        saccade = GazeFeatures.Saccade(random_gaze_positions(2))
+        saccade = TestSaccade(random_gaze_positions(2))
         ts, _ = saccade.positions.first
 
         new_step = aoi_scan_path.append_saccade(ts, saccade)
@@ -425,7 +449,7 @@ class TestAOIScanPathClass(unittest.TestCase):
         self.assertEqual(new_step, None)
 
         # Append fixation on B aoi
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
 
         new_step = aoi_scan_path.append_fixation(ts, fixation, 'Bar')
@@ -445,7 +469,7 @@ class TestAOIScanPathClass(unittest.TestCase):
         aoi_scan_path = GazeFeatures.AOIScanPath(['Foo', 'Bar'])
 
         # Append fixation on A aoi
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
 
         new_step = aoi_scan_path.append_fixation(ts, fixation, 'Foo')
@@ -455,7 +479,7 @@ class TestAOIScanPathClass(unittest.TestCase):
         self.assertEqual(new_step, None)
 
         # Append fixation on B aoi
-        fixation = GazeFeatures.Fixation(random_gaze_positions(10))
+        fixation = TestFixation(random_gaze_positions(10))
         ts, _ = fixation.positions.first
 
         # Check that aoi scan step creation fail when fixation is appened after another fixation