added new annotation types
This commit is contained in:
@@ -7,10 +7,13 @@ from uuid import UUID
|
||||
from data_store.repositories.annotation_repository import AnnotationRepository
|
||||
from data_store.dto import (
|
||||
Annotation,
|
||||
DistanceEnum,
|
||||
AngleEnum,
|
||||
PointOfViewEnum,
|
||||
ColourEnum,
|
||||
ContactEnum,
|
||||
DepthEnum,
|
||||
DistanceEnum,
|
||||
LightingEnum,
|
||||
PointOfViewEnum,
|
||||
)
|
||||
|
||||
|
||||
@@ -47,11 +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(
|
||||
distance=DistanceEnum.CLOSE,
|
||||
angle=AngleEnum.HIGH,
|
||||
point_of_view=PointOfViewEnum.FRONTAL,
|
||||
contact=ContactEnum.OFFER,
|
||||
)
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user