test_boson.py

import os
import time

import pytest

import flirpy.camera.boson
from flirpy.camera.boson import Boson

if Boson.find_video_device() is None:
    pytest.skip("Boson not connected, skipping tests", allow_module_level=True)


def test_open_boson():
    camera = Boson()
    camera.close()


@pytest.mark.skipif(os.name != "nt", reason="Skipping Windows-only test")
def test_capture_windows():
    with Boson() as camera:
        # Currently have no way of figuring this out
        res = camera.grab()

        assert res is not None
        assert len(res.shape) == 2
        assert res.dtype == "uint16"


@pytest.mark.skipif(os.name == "nt", reason="Skipping on Windows")
def test_capture_unix():
    with Boson() as camera:
        res = camera.grab()

        assert res is not None
        assert len(res.shape) == 2
        assert res.dtype == "uint16"


def test_get_serial():
    with Boson() as camera:
        assert camera.get_camera_serial() != 0


def test_get_sensor_serial():
    # Skipped until proper CRC is implemented
    with Boson() as camera:
        assert camera.get_sensor_serial() != 0


def test_get_firmware_revision():
    with Boson() as camera:
        rev = camera.get_firmware_revision()
        assert len(rev) == 3


def test_get_fpa_temperature():
    with Boson() as camera:
        # In principle this could be exactly zero, but it's quite unlikely.
        assert camera.get_fpa_temperature() != 0


def test_get_part_number():
    with Boson() as camera:
        pn = camera.get_part_number()
        assert pn != ""


def test_ffc_request():
    with Boson() as camera:
        n_frames = camera.get_frame_count()
        print(n_frames)
        camera.do_ffc()
        time.sleep(1)
        ffc_frame = camera.get_last_ffc_frame_count()
        assert (ffc_frame - n_frames) < 200


def test_ffc_temperature():
    with Boson() as camera:
        assert camera.get_last_ffc_temperature() > 0


def test_frame_count():
    with Boson() as camera:
        assert camera.get_frame_count() > 0


def test_ffc_frame_threshold():
    with Boson() as camera:
        thresh = 6000
        camera.set_ffc_frame_threshold(thresh)
        assert camera.get_ffc_frame_threshold() == thresh


def test_ffc_temperature_threshold():
    with Boson() as camera:
        temp_diff = 5.0
        camera.set_ffc_temperature_threshold(temp_diff)
        assert camera.get_ffc_temperature_threshold() == temp_diff


def test_ffc_mode_manual():
    with Boson() as camera:
        camera.set_ffc_manual()
        mode = camera.get_ffc_mode()
        assert mode == flirpy.camera.boson.FLR_BOSON_MANUAL_FFC


def test_ffc_mode_auto():
    with Boson() as camera:
        camera.set_ffc_auto()
        mode = camera.get_ffc_mode()
        assert mode == flirpy.camera.boson.FLR_BOSON_AUTO_FFC