Skip to content

Commit 1e6a0fe

Browse files
author
Val Brodsky
committed
Add another test
1 parent 34fe8f3 commit 1e6a0fe

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/integration/test_task_queue.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import time
22

33
from labelbox import Project
4-
from labelbox.schema.identifiables import GlobalKeys
4+
from labelbox.schema.identifiables import GlobalKeys, UniqueIds
55

66

77
def test_get_task_queue(project: Project):
@@ -31,7 +31,7 @@ def _validate_moved(project, queue_name, data_row_count):
3131
time.sleep(sleep_time)
3232

3333

34-
def test_move_to_task(configured_batch_project_with_label: Project):
34+
def test_move_to_task(configured_batch_project_with_label):
3535
project, _, data_row, _ = configured_batch_project_with_label
3636
task_queues = project.task_queues()
3737

@@ -45,3 +45,9 @@ def test_move_to_task(configured_batch_project_with_label: Project):
4545
project.move_data_rows_to_task_queue(GlobalKeys([data_row.global_key]),
4646
review_queue.uid)
4747
_validate_moved(project, "MANUAL_REWORK_QUEUE", 1)
48+
49+
review_queue = next(
50+
tq for tq in task_queues if tq.queue_type == "MANUAL_REVIEW_QUEUE")
51+
project.move_data_rows_to_task_queue(UniqueIds([data_row.uid]),
52+
review_queue.uid)
53+
_validate_moved(project, "MANUAL_REVIEW_QUEUE", 1)

0 commit comments

Comments
 (0)