Skip to content

Commit 66e8cb9

Browse files
author
Val Brodsky
committed
Update tests
1 parent 8441789 commit 66e8cb9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

libs/labelbox/tests/integration/test_labeling_dashboard.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
from datetime import datetime, timedelta
22
from labelbox.schema.labeling_service import LabelingServiceStatus
33
from labelbox.schema.search_filters import DateOperator, DateRange, DateRangeOperator, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkspaceFilter
4+
from labelbox.schema.ontology_kind import EditorTaskType
5+
from labelbox.schema.media_type import MediaType
46

57

68
def test_request_labeling_service_dashboard(rand_gen,
@@ -18,6 +20,9 @@ def test_request_labeling_service_dashboard(rand_gen,
1820
assert labeling_service_dashboard.status == LabelingServiceStatus.Missing
1921
assert labeling_service_dashboard.tasks_completed == 0
2022
assert labeling_service_dashboard.tasks_remaining == 0
23+
assert labeling_service_dashboard.media_type == MediaType.Conversational
24+
assert labeling_service_dashboard.editor_task_type == EditorTaskType.OfflineModelChatEvaluation
25+
assert labeling_service_dashboard.service_type == "Offline chat evaluation"
2126

2227
labeling_service_dashboard = [
2328
ld for ld in project.client.get_labeling_service_dashboards()

0 commit comments

Comments
 (0)