Newer
Older
sups_yolo / tests / unit / test_camera.py
"""Tests for camera abstractions."""

import numpy as np

from sups_yolo.camera.fake import FakeCameraSource


def test_fake_camera_generates_frames() -> None:
    camera = FakeCameraSource({"source": "data/datasets/fake/nonexistent"})
    camera.connect()
    frame = camera.get_frame()
    assert isinstance(frame, np.ndarray)
    assert frame.ndim == 3
    camera.disconnect()
    assert not camera.is_connected()