1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
#!/usr/bin/env python
import unittest
import random
import time
from argaze import GazeFeatures
from argaze.GazeAnalysis import DispersionBasedGazeMovementIdentifier
import numpy
def build_gaze_fixation(size: int, center: tuple, dispersion: float, start_time: float, min_time: float, max_time: float, validity: list = []):
""" Generate N TimeStampedGazePsoitions dispersed around a center point for testing purpose.
Timestamps are current time after random sleep (second).
GazePositions are random values.
"""
ts_gaze_positions = GazeFeatures.TimeStampedGazePositions()
for i in range(0, size):
# Check position validity
valid = True
if len(validity) > i:
valid = validity[i]
if valid:
# Edit gaze position
random_x = center[0] + dispersion * (random.random() - 0.5)
random_y = center[1] + dispersion * (random.random() - 0.5)
gaze_position = GazeFeatures.GazePosition((random_x, random_y))
else:
gaze_position = GazeFeatures.UnvalidGazePosition()
# Store gaze position
ts = time.time() - start_time
ts_gaze_positions[ts] = gaze_position
# Sleep a random time
sleep_time = random.random() * (max_time - min_time) + min_time
time.sleep(sleep_time)
return ts_gaze_positions
class TestDispersionBasedGazeMovementIdentifierClass(unittest.TestCase):
"""Test DispersionBasedGazeMovementIdentifier class."""
def test_fixation_identification(self):
"""Test DispersionBasedGazeMovementIdentifier fixation identification."""
size = 10
center = (0, 0)
dispersion = 10
start_time = time.time()
min_time = 0.01
max_time = 0.1
ts_gaze_positions = build_gaze_fixation(size, center, dispersion, start_time, min_time, max_time)
gaze_movement_identifier = DispersionBasedGazeMovementIdentifier.GazeMovementIdentifier(dispersion_threshold=dispersion, duration_threshold=min_time*2)
ts_fixations, ts_saccades, ts_status = gaze_movement_identifier.identify(ts_gaze_positions)
# Check result size
self.assertEqual(len(ts_fixations), 1)
self.assertEqual(len(ts_saccades), 0)
self.assertEqual(len(ts_status), size)
# Check fixation
ts, fixation = ts_fixations.pop_first()
self.assertEqual(len(fixation.positions.keys()), size)
self.assertLessEqual(fixation.dispersion, dispersion)
self.assertGreaterEqual(fixation.duration, size * min_time)
self.assertLessEqual(fixation.duration, size * max_time)
def test_fixation_and_saccade_identification(self):
"""Test DispersionBasedGazeMovementIdentifier fixation and saccade identification."""
size = 10
center_A = (0, 0)
center_B = (50, 50)
dispersion = 10
start_time = time.time()
min_time = 0.01
max_time = 0.1
ts_gaze_positions_A = build_gaze_fixation(size, center_A, dispersion, start_time, min_time, max_time)
ts_gaze_positions_B = build_gaze_fixation(size, center_B, dispersion, start_time, min_time, max_time)
ts_gaze_positions = ts_gaze_positions_A.append(ts_gaze_positions_B)
gaze_movement_identifier = DispersionBasedGazeMovementIdentifier.GazeMovementIdentifier(dispersion_threshold=dispersion, duration_threshold=min_time*2)
ts_fixations, ts_saccades, ts_status = gaze_movement_identifier.identify(ts_gaze_positions)
# Check result size
self.assertEqual(len(ts_fixations), 2)
self.assertEqual(len(ts_saccades), 1)
self.assertEqual(len(ts_status), size*2)
# Check first fixation
ts, fixation = ts_fixations.pop_first()
self.assertEqual(len(fixation.positions.keys()), size)
self.assertLessEqual(fixation.dispersion, dispersion)
self.assertGreaterEqual(fixation.duration, size * min_time)
self.assertLessEqual(fixation.duration, size * max_time)
# Check first saccade
ts, saccade = ts_saccades.pop_first()
self.assertEqual(len(saccade.positions.keys()), 2)
self.assertGreaterEqual(saccade.duration, min_time)
self.assertLessEqual(saccade.duration, max_time)
# Check second fixation
ts, fixation = ts_fixations.pop_first()
self.assertEqual(len(fixation.positions.keys()), size)
self.assertLessEqual(fixation.dispersion, dispersion)
self.assertGreaterEqual(fixation.duration, size * min_time)
self.assertLessEqual(fixation.duration, size * max_time)
def test_invalid_gaze_position(self):
"""Test DispersionBasedGazeMovementIdentifier fixation and saccade identification with invalid gaze position."""
size = 10
center = (0, 0)
dispersion = 10
start_time = time.time()
min_time = 0.01
max_time = 0.1
validity = [True, True, True, True, False, False, True, True, True, True]
ts_gaze_positions = build_gaze_fixation(size, center, dispersion, start_time, min_time, max_time, validity)
gaze_movement_identifier = DispersionBasedGazeMovementIdentifier.GazeMovementIdentifier(dispersion_threshold=dispersion, duration_threshold=min_time*2)
ts_fixations, ts_saccades, ts_status = gaze_movement_identifier.identify(ts_gaze_positions)
# Check result size
self.assertEqual(len(ts_fixations), 1)
self.assertEqual(len(ts_saccades), 0)
self.assertEqual(len(ts_status), size-2)
# Check fixation
ts, fixation = ts_fixations.pop_first()
self.assertEqual(len(fixation.positions.keys()), size-2)
self.assertLessEqual(fixation.dispersion, dispersion)
self.assertGreaterEqual(fixation.duration, size * min_time)
self.assertLessEqual(fixation.duration, size * max_time)
if __name__ == '__main__':
unittest.main()
|
