import pytest
|
|
from label_studio.tests.sdk.common import LABEL_CONFIG_AND_TASKS
|
|
pytestmark = pytest.mark.django_db
|
from label_studio_sdk.client import LabelStudio
|
|
|
def test_start_and_get_project(django_live_url, business_client):
|
ls = LabelStudio(base_url=django_live_url, api_key=business_client.api_key)
|
p = ls.projects.create(title='New Project', label_config=LABEL_CONFIG_AND_TASKS['label_config'])
|
|
project = ls.projects.get(id=p.id)
|
assert project
|
assert project.title == 'New Project'
|
|
ls.projects.update(id=project.id, title='Updated Project')
|
project = ls.projects.get(id=p.id)
|
assert project.title == 'Updated Project'
|
|
|
def test_delete_project(django_live_url, business_client):
|
ls = LabelStudio(base_url=django_live_url, api_key=business_client.api_key)
|
p = ls.projects.create(title='New Project', label_config=LABEL_CONFIG_AND_TASKS['label_config'])
|
|
project = ls.projects.get(id=p.id)
|
ls.projects.delete(id=project.id)
|
|
any_project_found = False
|
for project in ls.projects.list():
|
any_project_found = True
|
|
assert not any_project_found
|
|
|
def test_list_projects_with_params(django_live_url, business_client):
|
|
ls = LabelStudio(base_url=django_live_url, api_key=business_client.api_key)
|
ls.projects.create(title='Project 1', label_config=LABEL_CONFIG_AND_TASKS['label_config'])
|
ls.projects.create(title='Project 2', label_config=LABEL_CONFIG_AND_TASKS['label_config'])
|
|
projects = list(ls.projects.list())
|
assert len(projects) == 2
|
assert projects[0].title == 'Project 2'
|
assert projects[1].title == 'Project 1'
|
|
projects = list(ls.projects.list(filter='pinned_only'))
|
assert not projects
|
|
projects = list(ls.projects.list(include='id,title,pinned_at,created_at,created_by'))
|
assert projects[0].pinned_at is None
|
assert projects[0].created_at is not None
|
assert projects[0].created_by.email == business_client.user.email
|
