Skip to content

Commit 1826a42

Browse files
author
Val Brodsky
committed
Update integration test
1 parent f0ce9bc commit 1826a42

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
@@ -2,7 +2,7 @@
22
from labelbox.schema.labeling_service import LabelingServiceStatus
33
from labelbox.schema.ontology_kind import EditorTaskType
44
from labelbox.schema.media_type import MediaType
5-
from labelbox.schema.search_filters import RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter
5+
from labelbox.schema.search_filters import IntegerValue, RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, TaskCompletedCountFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter, TaskRemainingCountFilter
66

77

88
def test_request_labeling_service_dashboard(rand_gen,
@@ -91,3 +91,18 @@ def test_request_labeling_service_dashboard_filters(requested_labeling_service):
9191
labeling_service_dashboard = project.client.get_labeling_service_dashboards(
9292
).get_one()
9393
assert labeling_service_dashboard
94+
95+
task_done_count_filter = TaskCompletedCountFilter(
96+
operation=OperationType.TaskCompletedCount,
97+
value=IntegerValue(
98+
operator=RangeOperatorWithSingleValue.GreaterThanOrEqual, value=0))
99+
task_remaining_count_filter = TaskRemainingCountFilter(
100+
operation=OperationType.TaskRemainingCount,
101+
value=IntegerValue(
102+
operator=RangeOperatorWithSingleValue.GreaterThanOrEqual, value=0))
103+
104+
labeling_service_dashboard = [
105+
ld for ld in project.client.get_labeling_service_dashboards(
106+
search_query=[task_done_count_filter, task_remaining_count_filter])
107+
][0]
108+
assert labeling_service_dashboard is not None

0 commit comments

Comments
 (0)