diff --git a/tests/test_inspections.py b/tests/test_inspections.py index 534e70d..33c6aca 100644 --- a/tests/test_inspections.py +++ b/tests/test_inspections.py @@ -1,16 +1,12 @@ from alitra import Frame, Position -from robot_interface.models.mission.task import ( - RecordAudio, - TakeImage, - TakeThermalVideo, -) +from robot_interface.models.mission.task import RecordAudio, TakeImage, TakeThermalVideo from isar_robot import inspections target = Position(x=0, y=0, z=0, frame=Frame("robot")) -def test_create_image(): +def test_create_image() -> None: task_actions = TakeImage(target=target) inspection_image = inspections.create_image(task_actions) @@ -18,7 +14,7 @@ def test_create_image(): assert inspection_image.metadata.file_type == "jpg" -def test_create_video(): +def test_create_video() -> None: task_actions = TakeImage(target=target) inspection_video = inspections.create_video(task_actions) @@ -26,7 +22,7 @@ def test_create_video(): assert inspection_video.metadata.file_type == "mp4" -def test_create_thermal_video(): +def test_create_thermal_video() -> None: task_actions = TakeThermalVideo(target=target, duration=10) inspection_video = inspections.create_thermal_video(task_actions) @@ -35,7 +31,7 @@ def test_create_thermal_video(): assert inspection_video.metadata.duration == 10 -def test_create_audio(): +def test_create_audio() -> None: task_actions = RecordAudio(target=target, duration=10) inspection_recording = inspections.create_audio(task_actions) diff --git a/tests/test_telemetry.py b/tests/test_telemetry.py index 4a5baa6..cb71e91 100644 --- a/tests/test_telemetry.py +++ b/tests/test_telemetry.py @@ -1,14 +1,14 @@ from isar_robot.telemetry import _get_battery_level, _get_pressure_level -def test_get_battery_level(): +def test_get_battery_level() -> None: for _ in range(100): pressure_level: float = _get_battery_level() assert pressure_level >= 50 assert pressure_level <= 100 -def test_get_pressure_level(): +def test_get_pressure_level() -> None: for _ in range(100): pressure_level: float = _get_pressure_level() assert pressure_level >= 0.011