def test_get_predefined_label_for_scan_in_task__label_that_is_not_predefined(
prepare_environment: Any) -> None:
"""Test for fetching Predefined Label that is not predefined."""
# Step 1. Prepare a structure for the test
dataset = DatasetsRepository.add_new_dataset('KIDNEYS', 'Kidneys')
task = TasksRepository.add_task('MARK_KIDNEYS', 'Mark Kidneys',
'path/to/image', ['KIDNEYS'], [])
LabelTagsRepository.add_new_tag('EXAMPLE_TAG', 'Example Tag',
[LabelTool.RECTANGLE], task.id)
scan = ScansRepository.add_new_scan(dataset, 0)
user_id = UsersRepository.add_new_user(
User('user@medtagger', 'HASH', 'Admin', 'Admin'))
user = UsersRepository.get_user_by_id(user_id)
# Step 2. Add Label which is not predefined
LabelsRepository.add_new_label(scan.id,
task.key,
user,
LabelingTime(0),
is_predefined=False)
# Step 3. Check if there is no Predefined Label
predefined_label = LabelsRepository.get_predefined_label_for_scan_in_task(
scan, task)
assert not predefined_label
def test_get_predefined_brush_label_elements(prepare_environment: Any) -> None:
"""Test for fetching Predefined Brush Label Elements."""
# Step 1. Prepare a structure for the test
dataset = DatasetsRepository.add_new_dataset('KIDNEYS', 'Kidneys')
task = TasksRepository.add_task('MARK_KIDNEYS', 'Mark Kidneys',
'path/to/image', ['KIDNEYS'], [])
label_tag = LabelTagsRepository.add_new_tag('EXAMPLE_TAG', 'Example Tag',
[LabelTool.RECTANGLE], task.id)
scan = ScansRepository.add_new_scan(dataset, 3)
user_id = UsersRepository.add_new_user(
User('user@medtagger', 'HASH', 'Admin', 'Admin'))
user = UsersRepository.get_user_by_id(user_id)
# Step 2. Add Label with Brush Elements
label = LabelsRepository.add_new_label(scan.id,
task.key,
user,
LabelingTime(0),
is_predefined=True)
LabelsRepository.add_new_brush_label_element(label.id, 0, 0, 0, b'',
label_tag)
LabelsRepository.add_new_brush_label_element(label.id, 1, 0, 0, b'',
label_tag)
LabelsRepository.add_new_brush_label_element(label.id, 2, 0, 0, b'',
label_tag)
# Step 3. Check if there is is Predefined Label
brush_label_elements = LabelsRepository.get_predefined_brush_label_elements(
scan.id, task.id, 0, 3)
assert len(brush_label_elements) == 3
brush_label_elements = LabelsRepository.get_predefined_brush_label_elements(
scan.id, task.id, 0, 1)
assert len(brush_label_elements) == 1
def test_get_predefined_label_for_scan_in_task__predefined_label_for_given_task(
prepare_environment: Any) -> None:
"""Test for fetching Predefined Label only for specific Task."""
# Step 1. Prepare a structure for the test
dataset = DatasetsRepository.add_new_dataset('KIDNEYS', 'Kidneys')
task_left = TasksRepository.add_task('MARK_LEFT', 'Mark Left',
'path/to/image', ['KIDNEYS'], '', [],
[])
task_right = TasksRepository.add_task('MARK_RIGHT', 'Mark Right',
'path/to/image', ['KIDNEYS'], '', [],
[])
LabelTagsRepository.add_new_tag('EXAMPLE_TAG', 'Example Tag',
[LabelTool.RECTANGLE], task_left.id)
scan = ScansRepository.add_new_scan(dataset, 0)
user_id = UsersRepository.add_new_user(
User('user@medtagger', 'HASH', 'Admin', 'Admin'))
user = UsersRepository.get_user_by_id(user_id)
# Step 2. Add Labels for each Task
label_left = LabelsRepository.add_new_label(scan.id,
task_left.key,
user,
LabelingTime(0),
is_predefined=True)
label_right = LabelsRepository.add_new_label(scan.id,
task_right.key,
user,
LabelingTime(0),
is_predefined=True)
# Step 3. Check if there are these Predefined Labels
predefined_label = LabelsRepository.get_predefined_label_for_scan_in_task(
scan, task_left)
assert predefined_label
assert predefined_label.id == label_left.id
predefined_label = LabelsRepository.get_predefined_label_for_scan_in_task(
scan, task_right)
assert predefined_label
assert predefined_label.id == label_right.id
def add_label(scan_id: ScanID, task_key: str, elements: List[Dict], # pylint: disable-msg=too-many-arguments
files: Dict[str, bytes], labeling_time: LabelingTime, comment: str = None,
is_predefined: bool = False) -> Label:
"""Add label to given scan.
:param scan_id: ID of a given scan
:param task_key: Key of Task
:param elements: List of JSONs describing elements for a single label
:param files: mapping of uploaded files (name and content)
:param labeling_time: time in seconds that user spent on labeling
:param comment: (optional) comment describing a label
:param is_predefined: (optional) mark such Label as predefined to show on Labeling Page
:return: Label object
"""
user = get_current_user()
try:
label = LabelsRepository.add_new_label(scan_id, task_key, user, labeling_time, comment, is_predefined)
except IntegrityError:
raise NotFoundException('Could not find Scan for that id!')
for element in elements:
add_label_element(element, label.id, files)
return label
def prepare_empty_label(scan: models.Scan, user: models.User) -> models.Label:
"""Create empty Label for given Scan and User."""
return LabelsRepository.add_new_label(scan.id, 'MARK_KIDNEYS', user,
LabelingTime(0.0))
