aboutsummaryrefslogtreecommitdiff
path: root/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
diff options
context:
space:
mode:
authorThéo de la Hogue2023-07-03 14:53:40 +0200
committerThéo de la Hogue2023-07-03 14:53:40 +0200
commit3d66f3e2d58b7038cdfdf272f2d813e24b730243 (patch)
tree912f8c70933f1f4b3d0097b93f6089b5f7914f23 /src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
parent25554ea7710c35107f0904532df093f9c15f74ae (diff)
downloadargaze-3d66f3e2d58b7038cdfdf272f2d813e24b730243.zip
argaze-3d66f3e2d58b7038cdfdf272f2d813e24b730243.tar.gz
argaze-3d66f3e2d58b7038cdfdf272f2d813e24b730243.tar.bz2
argaze-3d66f3e2d58b7038cdfdf272f2d813e24b730243.tar.xz
Updating GazeAnalysis test.
Diffstat (limited to 'src/argaze.test/GazeAnalysis/NearestNeighborIndex.py')
-rw-r--r--src/argaze.test/GazeAnalysis/NearestNeighborIndex.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py b/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
index fb7d4ec..abd6b2c 100644
--- a/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
+++ b/src/argaze.test/GazeAnalysis/NearestNeighborIndex.py
@@ -21,19 +21,20 @@ class TestScanPathAnalyzer(unittest.TestCase):
def test_analyze(self):
"""Test analyze."""
- nni_analyzer = NearestNeighborIndex.ScanPathAnalyzer()
-
screen_dimension = (100, 100)
+
+ nni_analyzer = NearestNeighborIndex.ScanPathAnalyzer(size=screen_dimension)
+
scan_path = GazeFeaturesTest.build_scan_path(6, screen_dimension)
# Check aoi scan path
self.assertEqual(len(scan_path), 6)
- nni = nni_analyzer.analyze(scan_path, screen_dimension)
+ nni_analyzer.analyze(scan_path)
# Check NNI
- self.assertGreaterEqual(nni, 0)
- self.assertLessEqual(nni, 1)
+ self.assertGreaterEqual(nni_analyzer.nearest_neighbor_index, 0)
+ self.assertLessEqual(nni_analyzer.nearest_neighbor_index, 1)
if __name__ == '__main__':