code quality fixes
This commit is contained in:
@@ -2,14 +2,10 @@
|
|||||||
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import shutil
|
|
||||||
import tempfile
|
|
||||||
from pathlib import Path
|
|
||||||
import json
|
import json
|
||||||
from collections.abc import Generator
|
from collections.abc import Generator
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
import pandas as pd
|
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
import pytest
|
import pytest
|
||||||
import structlog
|
import structlog
|
||||||
|
|||||||
@@ -50,21 +50,21 @@ class TestAnnotationRepository(unittest.TestCase):
|
|||||||
cls.job_id = UUID("d87022f8-4169-4d69-8512-bbd65cc1cb23")
|
cls.job_id = UUID("d87022f8-4169-4d69-8512-bbd65cc1cb23")
|
||||||
cls.bad_job_id = "not-a-uuid"
|
cls.bad_job_id = "not-a-uuid"
|
||||||
cls.annotation = Annotation(
|
cls.annotation = Annotation(
|
||||||
angle=AngleEnum.HIGH,
|
angle=AngleEnum.HIGH,
|
||||||
angle_uncertainty=True,
|
angle_uncertainty=True,
|
||||||
colour=ColourEnum.HIGH,
|
colour=ColourEnum.HIGH,
|
||||||
colour_uncertainty=False,
|
colour_uncertainty=False,
|
||||||
contact=ContactEnum.OFFER,
|
contact=ContactEnum.OFFER,
|
||||||
contact_uncertainty=True,
|
contact_uncertainty=True,
|
||||||
depth=DepthEnum.LOW,
|
depth=DepthEnum.LOW,
|
||||||
depth_uncertainty=False,
|
depth_uncertainty=False,
|
||||||
distance=DistanceEnum.CLOSE,
|
distance=DistanceEnum.CLOSE,
|
||||||
distance_uncertainty=True,
|
distance_uncertainty=True,
|
||||||
lighting=LightingEnum.MEDIUM,
|
lighting=LightingEnum.MEDIUM,
|
||||||
lighting_uncertainty=False,
|
lighting_uncertainty=False,
|
||||||
point_of_view=PointOfViewEnum.FRONTAL,
|
point_of_view=PointOfViewEnum.FRONTAL,
|
||||||
point_of_view_uncertainty=True,
|
point_of_view_uncertainty=True,
|
||||||
)
|
)
|
||||||
cls.bad_annotation = {"not_an_annotation": "irrelevant value"}
|
cls.bad_annotation = {"not_an_annotation": "irrelevant value"}
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|||||||
Reference in New Issue
Block a user