From 1ea84dca42623cf44e14334407ab532a896c97cb Mon Sep 17 00:00:00 2001 From: Maxim Zhiltsov Date: Thu, 9 Jan 2025 14:28:50 +0200 Subject: [PATCH] Update tests --- .../services/test_validation_service.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/packages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py b/packages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py index 40c0f2cfd4..4894defb27 100644 --- a/packages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py +++ b/packages/examples/cvat/recording-oracle/tests/integration/services/test_validation_service.py @@ -808,6 +808,13 @@ def test_can_exclude_bad_gt_for_each_label_separately(self, session: Session): frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)], ) + common_lock_es.enter_context( + mock.patch( + "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings", + return_value=cvat_api.QualitySettings(target_metric="accuracy"), + ) + ) + def patched_get_task_labels(task_id: int): return [manifest.annotation.labels[0].nodes[task_id - 1]] @@ -1009,6 +1016,13 @@ def test_can_complete_if_not_enough_gt_left_in_task( frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)], ) + common_lock_es.enter_context( + mock.patch( + "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings", + return_value=cvat_api.QualitySettings(target_metric="accuracy"), + ) + ) + def patched_get_task_labels(task_id: int): return [manifest.annotation.labels[0].nodes[task_id - 1]] @@ -1185,6 +1199,13 @@ def test_can_complete_if_not_enough_gt_left_in_task( frames=[SimpleNamespace(name=f"frame_{i}.jpg") for i in range(frame_count)], ) + common_lock_es.enter_context( + mock.patch( + "src.handlers.process_intermediate_results.cvat_api.get_task_quality_settings", + return_value=cvat_api.QualitySettings(target_metric="accuracy"), + ) + ) + def patched_get_task_labels(task_id: int): return [manifest.annotation.labels[0].nodes[task_id - 1]]