Improve docs for hardware specific tests
This commit is contained in:
parent
11c8f47e20
commit
a43a82213d
4 changed files with 61 additions and 11 deletions
|
|
@ -1,11 +1,13 @@
|
|||
from labthings_picamera2 import StreamingPiCamera2
|
||||
from labthings_fastapi.server import ThingServer
|
||||
from labthings_fastapi.client import ThingClient
|
||||
from fastapi.testclient import TestClient
|
||||
from PIL import Image
|
||||
import numpy as np
|
||||
from pytest import fixture
|
||||
|
||||
from labthings_fastapi.server import ThingServer
|
||||
from labthings_fastapi.client import ThingClient
|
||||
|
||||
from openflexure_microscope_server.things.camera.picamera import StreamingPiCamera2
|
||||
|
||||
|
||||
@fixture(scope="module")
|
||||
def client():
|
||||
|
|
@ -17,10 +19,17 @@ def client():
|
|||
|
||||
|
||||
def test_calibration(client):
|
||||
"""
|
||||
Check that full auto calibrate completes without an exception
|
||||
"""
|
||||
client.full_auto_calibrate()
|
||||
|
||||
|
||||
def test_jpeg_and_array(client):
|
||||
"""
|
||||
Check that grabbing a jpeg from the stream results in the same size
|
||||
image as a array capture or a jpeg capture.
|
||||
"""
|
||||
blob = client.grab_jpeg()
|
||||
mjpeg_frame = Image.open(blob.open())
|
||||
assert mjpeg_frame
|
||||
|
|
|
|||
|
|
@ -5,12 +5,18 @@ from fastapi.testclient import TestClient
|
|||
|
||||
from labthings_fastapi.server import ThingServer
|
||||
from labthings_fastapi.client import ThingClient
|
||||
from labthings_picamera2.thing import StreamingPiCamera2
|
||||
|
||||
from openflexure_microscope_server.things.camera.picamera import StreamingPiCamera2
|
||||
|
||||
logging.basicConfig(level=logging.DEBUG)
|
||||
|
||||
|
||||
def test_exposure_time_drift():
|
||||
"""
|
||||
Capture 10 full resolution images and check that the exposure time remains constant
|
||||
|
||||
This confirms that automatic exposure time adjustment is fully turned off
|
||||
"""
|
||||
cam = StreamingPiCamera2()
|
||||
server = ThingServer()
|
||||
server.add_thing(cam, "/camera/")
|
||||
|
|
|
|||
|
|
@ -5,12 +5,16 @@ import numpy as np
|
|||
|
||||
from labthings_fastapi.server import ThingServer
|
||||
from labthings_fastapi.client import ThingClient
|
||||
from labthings_picamera2.thing import StreamingPiCamera2
|
||||
|
||||
from openflexure_microscope_server.things.camera.picamera import StreamingPiCamera2
|
||||
|
||||
logging.basicConfig(level=logging.DEBUG)
|
||||
|
||||
|
||||
def test_sensor_mode():
|
||||
"""
|
||||
Test capturing raw arrays in two different sensor modes
|
||||
"""
|
||||
cam = StreamingPiCamera2()
|
||||
server = ThingServer()
|
||||
server.add_thing(cam, "/camera/")
|
||||
|
|
|
|||
|
|
@ -1,23 +1,30 @@
|
|||
"""
|
||||
Tests that check that a tuning file can be reloaded
|
||||
"""
|
||||
|
||||
import os
|
||||
from picamera2 import Picamera2
|
||||
from labthings_picamera2 import recalibrate_utils
|
||||
|
||||
import pytest
|
||||
from picamera2 import Picamera2
|
||||
|
||||
from openflexure_microscope_server.things.camera import recalibrate_utils
|
||||
|
||||
|
||||
MODEL = Picamera2.global_camera_info()[0]["Model"]
|
||||
|
||||
|
||||
def check_camera_available():
|
||||
assert len(Picamera2.global_camera_info()) >= 1
|
||||
|
||||
|
||||
def load_default_tuning():
|
||||
"""
|
||||
Return the default tuning file for the connected camera.
|
||||
"""
|
||||
fname = f"{MODEL}.json"
|
||||
return Picamera2.load_tuning_file(fname)
|
||||
|
||||
|
||||
def generate_bad_tuning():
|
||||
"""
|
||||
Return a tuning file with an invalid version number to force an error when loaded.
|
||||
"""
|
||||
default_tuning = load_default_tuning()
|
||||
bad_tuning = default_tuning.copy()
|
||||
bad_tuning["version"] = 999
|
||||
|
|
@ -25,6 +32,14 @@ def generate_bad_tuning():
|
|||
|
||||
|
||||
def print_tuning(read_file=False):
|
||||
"""
|
||||
Print the path of the default tuning file from the the environment variable.
|
||||
|
||||
:param read_file: Boolean, set true to also print the file contents.
|
||||
|
||||
This is useful for debuging. As PyTest suppresses the printing by default the
|
||||
-s option is needed when running pylint to see this.
|
||||
"""
|
||||
key = "LIBCAMERA_RPI_TUNING_FILE"
|
||||
if key in os.environ:
|
||||
print(f"Tuning file environment variable: {os.environ[key]}")
|
||||
|
|
@ -36,6 +51,16 @@ def print_tuning(read_file=False):
|
|||
|
||||
|
||||
def _test_bad_tuning_after_good_tuning(configure):
|
||||
"""
|
||||
Load the default tuning file into the camera, re-load with a broken tuning file,
|
||||
check it errors. Finally check the default tuning file will load again afterwards.
|
||||
|
||||
:param configure: Boolean, set true to configure the camera on initial loading
|
||||
|
||||
This test checks that:
|
||||
recalibrate_utils.recreate_camera_manager() is working as expected. As the default
|
||||
PiCamera2 behaviour does not expect the tuning file to be reloaded.
|
||||
"""
|
||||
bad_tuning = generate_bad_tuning()
|
||||
default_tuning = load_default_tuning()
|
||||
print_tuning()
|
||||
|
|
@ -61,6 +86,12 @@ def _test_bad_tuning_after_good_tuning(configure):
|
|||
del cam
|
||||
|
||||
|
||||
# Note: The tests below are marked with
|
||||
# @pytest.mark.filterwarnings("ignore: Exception ignored")
|
||||
# as the picamera will throw an exception ignored warning when deleting
|
||||
# the picamera object. This is expected
|
||||
|
||||
|
||||
@pytest.mark.filterwarnings("ignore: Exception ignored")
|
||||
def test_bad_tuning_after_good_tuning_noconfigure():
|
||||
_test_bad_tuning_after_good_tuning(False)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue