Skip to content

Commit bc37352

Browse files
author
Val Brodsky
committed
Update integration test
1 parent 1b65f12 commit bc37352

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

libs/labelbox/tests/integration/test_labeling_dashboard.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from datetime import datetime, timedelta
22
from labelbox.schema.labeling_service import LabelingServiceStatus
3-
from labelbox.schema.search_filters import RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter
3+
from labelbox.schema.search_filters import IntegerValue, RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, TaskCompletedCountFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter, TaskRemainingCountFilter
44

55

66
def test_request_labeling_service_dashboard(rand_gen,
@@ -83,3 +83,18 @@ def test_request_labeling_service_dashboard_filters(requested_labeling_service):
8383
]
8484
assert len(labeling_service_dashboard) == 0
8585
assert labeling_service_dashboard == []
86+
87+
task_done_count_filter = TaskCompletedCountFilter(
88+
operation=OperationType.TaskCompletedCount,
89+
value=IntegerValue(
90+
operator=RangeOperatorWithSingleValue.GreaterThanOrEqual, value=0))
91+
task_remaining_count_filter = TaskRemainingCountFilter(
92+
operation=OperationType.TaskRemainingCount,
93+
value=IntegerValue(
94+
operator=RangeOperatorWithSingleValue.GreaterThanOrEqual, value=0))
95+
96+
labeling_service_dashboard = [
97+
ld for ld in project.client.get_labeling_service_dashboards(
98+
search_query=[task_done_count_filter, task_remaining_count_filter])
99+
][0]
100+
assert labeling_service_dashboard is not None

0 commit comments

Comments
 (0)