wehub-resource-sync
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
# Copyright (C) CVAT.ai Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from cvat_sdk.api_client import models
|
|
from cvat_sdk.core.proxies.tasks import Task
|
|
|
|
|
|
@pytest.mark.parametrize("target_type", ["task", "job"])
|
|
@pytest.mark.parametrize(
|
|
("import_mode", "expected_shape_count"),
|
|
[
|
|
("replace", 1),
|
|
("append", 2),
|
|
],
|
|
)
|
|
def test_import_annotations_respects_import_mode(
|
|
fxt_new_task: Task,
|
|
target_type: str,
|
|
import_mode: str,
|
|
expected_shape_count: int,
|
|
tmp_path: Path,
|
|
):
|
|
labels = fxt_new_task.get_labels()
|
|
target = fxt_new_task if target_type == "task" else fxt_new_task.get_jobs()[0]
|
|
|
|
target.set_annotations(
|
|
models.LabeledDataRequest(
|
|
shapes=[
|
|
models.LabeledShapeRequest(
|
|
frame=0,
|
|
label_id=labels[0].id,
|
|
type="rectangle",
|
|
points=[1, 1, 2, 2],
|
|
),
|
|
],
|
|
)
|
|
)
|
|
|
|
original_annotations = target.get_annotations()
|
|
assert len(original_annotations.shapes) == 1
|
|
|
|
exported_dataset = target.export_dataset(
|
|
format_name="CVAT for images 1.1",
|
|
filename=tmp_path / f"{target_type}_annotations.zip",
|
|
include_images=False,
|
|
)
|
|
|
|
target.import_annotations(
|
|
format_name="CVAT 1.1",
|
|
filename=exported_dataset,
|
|
import_mode=import_mode,
|
|
)
|
|
|
|
imported_annotations = target.get_annotations()
|
|
assert len(imported_annotations.shapes) == expected_shape_count
|