diff options
Diffstat (limited to 'src/argaze.test/PupilAnalysis')
| -rw-r--r-- | src/argaze.test/PupilAnalysis/WorkloadIndex.py | 46 | ||||
| -rw-r--r-- | src/argaze.test/PupilAnalysis/__init__.py | 0 |
2 files changed, 46 insertions, 0 deletions
diff --git a/src/argaze.test/PupilAnalysis/WorkloadIndex.py b/src/argaze.test/PupilAnalysis/WorkloadIndex.py new file mode 100644 index 0000000..98a1e45 --- /dev/null +++ b/src/argaze.test/PupilAnalysis/WorkloadIndex.py @@ -0,0 +1,46 @@ +#!/usr/bin/env python + +import unittest +import math + +from argaze import PupilFeatures +from argaze.PupilAnalysis import WorkloadIndex + +class TestWorkloadIndexClass(unittest.TestCase): + """Test WorkloadIndex class.""" + + def test_analysis(self): + """Test WorkloadIndex analysis.""" + + ts_pupil_diameters = { + 0: PupilFeatures.PupilDiameter(1.), + 1: PupilFeatures.PupilDiameter(1.1), + 2: PupilFeatures.PupilDiameter(1.2), + 3: PupilFeatures.PupilDiameter(1.3), + 4: PupilFeatures.PupilDiameter(1.2), + 5: PupilFeatures.PupilDiameter(1.1), + 6: PupilFeatures.PupilDiameter(1.), + 7: PupilFeatures.PupilDiameter(0.9), + 8: PupilFeatures.PupilDiameter(0.8), + 9: PupilFeatures.PupilDiameter(0.7) + } + + pupil_diameter_analyzer = WorkloadIndex.PupilDiameterAnalyzer(reference=PupilFeatures.PupilDiameter(1.), period=3) + ts_analysis = pupil_diameter_analyzer.browse(PupilFeatures.TimeStampedPupilDiameters(ts_pupil_diameters)) + + # Check result size + self.assertEqual(len(ts_analysis), 3) + + # Check each workload index + ts_1, analysis_1 = ts_analysis.pop_first() + self.assertTrue(math.isclose(analysis_1, 0.1, abs_tol=1e-2)) + + ts_2, analysis_2 = ts_analysis.pop_first() + self.assertTrue(math.isclose(analysis_2, 0.2, abs_tol=1e-2)) + + ts_3, analysis_3 = ts_analysis.pop_first() + self.assertTrue(math.isclose(analysis_3, 0.1, abs_tol=1e-2)) + +if __name__ == '__main__': + + unittest.main()
\ No newline at end of file diff --git a/src/argaze.test/PupilAnalysis/__init__.py b/src/argaze.test/PupilAnalysis/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/src/argaze.test/PupilAnalysis/__init__.py |