Skip to content

Commit cfe7d0c

Browse files
committed
Update test
1 parent 7ba272c commit cfe7d0c

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

libs/labelbox/tests/integration/test_task_queue.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,18 +11,31 @@ def test_get_task_queue(project: Project):
1111
assert review_queue
1212

1313

14-
def test_get_overview(project: Project):
14+
def test_get_overview_no_details(project: Project):
1515
po = project.get_overview()
1616

1717
assert isinstance(po.to_label, int)
18-
assert isinstance(po.in_review, dict)
18+
assert isinstance(po.in_review, int)
1919
assert isinstance(po.in_rework, int)
2020
assert isinstance(po.skipped, int)
2121
assert isinstance(po.done, int)
2222
assert isinstance(po.issues, int)
2323
assert isinstance(po.labeled, int)
2424
assert isinstance(po.all_in_data_rows, int)
2525

26+
def test_get_overview_with_details(project: Project):
27+
po = project.get_overview(details=True)
28+
29+
assert isinstance(po.to_label, int)
30+
assert isinstance(po.in_review["data"], list)
31+
assert isinstance(po.in_review["total"], int)
32+
assert isinstance(po.in_rework["data"], list)
33+
assert isinstance(po.in_rework["total"], int)
34+
assert isinstance(po.skipped, int)
35+
assert isinstance(po.done, int)
36+
assert isinstance(po.issues, int)
37+
assert isinstance(po.labeled, int)
38+
assert isinstance(po.all_in_data_rows, int)
2639

2740
def _validate_moved(project, queue_name, data_row_count):
2841
timeout_seconds = 30

0 commit comments

Comments
 (0)