tests/integration/conftest.py

@@ -2,14 +2,10 @@
 
 import logging
 import os
-import shutil
-import tempfile
-from pathlib import Path
 import json
 from collections.abc import Generator
 from uuid import UUID
 from io import BytesIO
-import pandas as pd
 from PIL import Image
 import pytest
 import structlog

tests/unit/test_annotation_repository.py

@@ -50,21 +50,21 @@ class TestAnnotationRepository(unittest.TestCase):
         cls.job_id = UUID("d87022f8-4169-4d69-8512-bbd65cc1cb23")
         cls.bad_job_id = "not-a-uuid"
         cls.annotation = Annotation(
-        angle=AngleEnum.HIGH,
-        angle_uncertainty=True,
-        colour=ColourEnum.HIGH,
-        colour_uncertainty=False,
-        contact=ContactEnum.OFFER,
-        contact_uncertainty=True,
-        depth=DepthEnum.LOW,
-        depth_uncertainty=False,
-        distance=DistanceEnum.CLOSE,
-        distance_uncertainty=True,
-        lighting=LightingEnum.MEDIUM,
-        lighting_uncertainty=False,
-        point_of_view=PointOfViewEnum.FRONTAL,
-        point_of_view_uncertainty=True,
-    )
+            angle=AngleEnum.HIGH,
+            angle_uncertainty=True,
+            colour=ColourEnum.HIGH,
+            colour_uncertainty=False,
+            contact=ContactEnum.OFFER,
+            contact_uncertainty=True,
+            depth=DepthEnum.LOW,
+            depth_uncertainty=False,
+            distance=DistanceEnum.CLOSE,
+            distance_uncertainty=True,
+            lighting=LightingEnum.MEDIUM,
+            lighting_uncertainty=False,
+            point_of_view=PointOfViewEnum.FRONTAL,
+            point_of_view_uncertainty=True,
+        )
         cls.bad_annotation = {"not_an_annotation": "irrelevant value"}
 
     @classmethod