Added basic tests

.gitattributes

@@ -0,0 +1 @@
+tests/data/demo_recording filter=lfs diff=lfs merge=lfs -text

src/pupil_labs/neon_recording/__init__.py

@@ -3,7 +3,7 @@
 
 import structlog
 
-from .neon_recording import load
+from .neon_recording import load, NeonRecording
 
 try:
     from importlib.metadata import PackageNotFoundError, version
@@ -34,4 +34,4 @@
     ],
 )
 
-__all__ = ["__version__", "load"]
+__all__ = ["__version__", "load", "NeonRecording"]

tests/test_neon_recording.py

@@ -0,0 +1,228 @@
+from pathlib import Path
+
+import numpy as np
+import pytest
+
+import pupil_labs.neon_recording as nr
+
+
+@pytest.fixture
+def rec():
+    rec_dir = Path("tests/data/demo_recording")
+    return nr.load(rec_dir)
+
+
+def test_gaze(rec: nr.NeonRecording):
+    sensor = rec.gaze
+    ts = sensor.ts
+    assert len(sensor) == 1951
+
+    for a in list(sensor.sample(ts[1950]))[0]:
+        print(f"{a},")
+
+    target_0 = (
+        1726825502.652337,
+        740.6018676757812,
+        685.78662109375,
+    )
+    for a, b in zip(list(sensor.sample(ts[0]))[0], target_0):
+        assert a == b
+
+    target_100 = (
+        1726825503.3129618,
+        766.157958984375,
+        705.6735229492188,
+    )
+    for a, b in zip(list(sensor.sample(ts[100]))[0], target_100):
+        assert a == b
+
+    target_1950 = (
+        1726825512.576638,
+        1081.5859375,
+        1174.9180908203125,
+    )
+    for a, b in zip(list(sensor.sample(ts[1950]))[0], target_1950):
+        assert a == b
+
+
+def test_eye_state(rec: nr.NeonRecording):
+    sensor = rec.eye_state
+    ts = sensor.ts
+    assert len(sensor) == 1754
+
+    target_0 = (
+        1726825503.7984638,
+        5.107705593109131,
+        -29.4140625,
+        10.517578125,
+        -33.057861328125,
+        0.07107485830783844,
+        0.24335210025310516,
+        0.9673304557800293,
+        5.657186508178711,
+        33.5498046875,
+        12.3046875,
+        -33.194580078125,
+        -0.2051464319229126,
+        0.23124463856220245,
+        0.9510209560394287,
+    )
+    for a, b in zip(list(sensor.sample(ts[0]))[0], target_0):
+        assert a == b
+
+    target_100 = (
+        1726825504.298967,
+        5.102882385253906,
+        -29.677734375,
+        10.625,
+        -32.87841796875,
+        0.10377761721611023,
+        0.2530110478401184,
+        0.9618813395500183,
+        5.733665943145752,
+        33.291015625,
+        12.333984375,
+        -33.15185546875,
+        -0.17958272993564606,
+        0.24260810017585754,
+        0.9533579349517822,
+    )
+    for a, b in zip(list(sensor.sample(ts[100]))[0], target_100):
+        assert a == b
+
+    target_1753 = (
+        1726825512.576638,
+        5.204510688781738,
+        -28.88671875,
+        10.771484375,
+        -31.007080078125,
+        0.35248205065727234,
+        0.6131130456924438,
+        0.7069998383522034,
+        4.113816261291504,
+        33.8720703125,
+        12.67578125,
+        -35.211181640625,
+        0.10800117999315262,
+        0.6311267614364624,
+        0.7681241631507874,
+    )
+    for a, b in zip(list(sensor.sample(ts[1753]))[0], target_1753):
+        assert a == b
+
+
+def test_imu(rec: nr.NeonRecording):
+    sensor = rec.imu
+    ts = sensor.ts
+    assert len(sensor) == 1094
+
+    target_0 = (
+        1726825503.025795,
+        -1.39617919921875,
+        -5.626678466796875,
+        2.2525787353515625,
+        -0.0581054612994194,
+        -0.486328125,
+        0.8964843153953552,
+        -41.28612518310547,
+        44.89975357055664,
+        -28.282575607299805,
+        0.8715572357177734,
+        -0.22864609956741333,
+        -0.0807524248957634,
+        0.42613154649734497,
+    )
+    for a, b in zip(list(sensor.sample(ts[0]))[0], target_0):
+        assert a == b
+
+    target_100 = (
+        1726825503.8969738,
+        -4.8770904541015625,
+        -7.3375701904296875,
+        -2.0809173583984375,
+        -0.1308593600988388,
+        -0.4711913764476776,
+        0.85546875,
+        -37.98215103149414,
+        38.5346794128418,
+        -19.76430320739746,
+        0.897792398929596,
+        -0.2490871101617813,
+        -0.04740327596664429,
+        0.3601073622703552,
+    )
+    for a, b in zip(list(sensor.sample(ts[100]))[0], target_100):
+        assert a == b
+
+    target_1093 = (
+        1726825512.557629,
+        -7.07244873046875,
+        3.467559814453125,
+        -4.4002532958984375,
+        -0.0507812462747097,
+        -0.52880859375,
+        0.8588866591453552,
+        -41.224647521972656,
+        37.80641174316406,
+        -24.91761016845703,
+        0.8892564177513123,
+        -0.2597949504852295,
+        -0.07967071235179901,
+        0.3679431080818176,
+    )
+    for a, b in zip(list(sensor.sample(ts[1093]))[0], target_1093):
+        assert a == b
+
+
+def test_events(rec: nr.NeonRecording):
+    sensor = rec.events
+    ts = sensor.ts
+    assert len(sensor) == 2
+
+    target_0 = (1726825501.5430002, "recording.begin")
+    for a, b in zip(list(sensor.sample(ts[0]))[0], target_0):
+        assert a == b
+
+    target_1 = (1726825512.556, "recording.end")
+    for a, b in zip(list(sensor.sample(ts[1]))[0], target_1):
+        assert a == b
+
+
+def test_scene(rec: nr.NeonRecording):
+    sensor = rec.scene
+    ts = sensor.ts
+    assert len(sensor) == 306
+
+    target_0 = 150.85475729166666
+    a = list(sensor.sample(ts[0:1]))[0]
+    assert np.mean(a.bgr) == target_0
+
+    target_100 = 150.60516822916668
+    a = list(sensor.sample(ts[100:101]))[0]
+    assert np.mean(a.bgr) == target_100
+
+    target_305 = 146.23578055555555
+    a = list(sensor.sample(ts[305:306]))[0]
+    assert np.mean(a.bgr) == target_305
+
+
+def test_eye(rec: nr.NeonRecording):
+    sensor = rec.eye
+    ts = sensor.ts
+    assert len(sensor) == 1951
+
+    for i in [0, 100, 1950]:
+        a = list(sensor.sample(ts[i : i + 1]))[0]
+        print(f"{np.mean(a.bgr)},")
+
+    target_0 = 122.28241644965277
+    a = list(sensor.sample(ts[0:1]))[0]
+    assert np.mean(a.bgr) == target_0
+
+    target_100 = 121.9274269386574
+    a = list(sensor.sample(ts[100:101]))[0]
+    assert np.mean(a.bgr) == target_100
+
+    target_1950 = 134.1152298538773
+    a = list(sensor.sample(ts[1950:1951]))[0]
+    assert np.mean(a.bgr) == target_1950