Update tests

label_studio/tests/conftest.py

@@ -34,6 +34,8 @@
     print('\n\n !!! Please, pip install pytest-env \n\n')
     exit(-100)
 
+from label_studio.tests.sdk.fixtures import *  # noqa: F403
+
 from .utils import (
     azure_client_mock,
     create_business,

label_studio/tests/sdk/fixtures.py

@@ -0,0 +1,29 @@
+import pytest
+from label_studio_sdk.client import LabelStudio
+from label_studio_sdk.data_manager import Column, Filters, Operator, Type
+
+from .common import LABEL_CONFIG_AND_TASKS
+
+
+@pytest.fixture
+def test_project_with_view(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)
+
+    task_data = [{'data': {'my_text': 'Test task ' + str(i)}} for i in range(10)]
+    ls.projects.import_tasks(id=project.id, request=task_data)
+    orig_tasks = []
+    for task in ls.tasks.list(project=project.id):
+        orig_tasks.append(task)
+
+    filters = Filters.create(
+        Filters.OR,
+        [Filters.item(Column.id, Operator.EQUAL, Type.Number, Filters.value(t.id)) for t in orig_tasks[::2]],
+    )
+
+    view = ls.views.create(
+        project=project.id, data=dict(title='Test View', filters=filters, ordering=['-' + Column.id])
+    )
+    return ls, project, orig_tasks, view

label_studio/tests/sdk/test_export.py

@@ -35,3 +35,24 @@ def test_direct_export(test_project):
     df = ls.projects.exports.as_pandas(project.id)
     assert isinstance(df, pd.DataFrame)
     assert len(df) == 1
+
+    # Test low level export - import new task without annotations
+    ls.projects.import_tasks(
+        id=project.id,
+        request={
+            'data': {
+                'my_text': 'Opossums are great',
+                'ref_id': 456,
+                'meta_info': {'timestamp': '2020-03-09 18:15:28.212882', 'location': 'North Pole'},
+            }
+        },
+    )
+    data = ls.projects.exports.download_sync(project.id, download_all_tasks=False)
+    from label_studio_sdk.projects.exports.client_ext import _bytestream_to_json
+
+    assert len(_bytestream_to_json(data)) == 1
+
+    data = ls.projects.exports.download_sync(project.id, download_all_tasks=True)
+    from label_studio_sdk.projects.exports.client_ext import _bytestream_to_json
+
+    assert len(_bytestream_to_json(data)) == 2

label_studio/tests/sdk/test_views.py

@@ -32,27 +32,13 @@ def test_create_view(django_live_url, business_client):
     }
 
 
-def test_get_tasks_from_view(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)
-
-    task_data = [{'data': {'my_text': 'Test task ' + str(i)}} for i in range(10)]
-    ls.projects.import_tasks(id=project.id, request=task_data)
-    orig_tasks = []
-    for task in ls.tasks.list(project=project.id):
-        orig_tasks.append(task)
-
-    filters = Filters.create(
-        Filters.OR,
-        [Filters.item(Column.id, Operator.EQUAL, Type.Number, Filters.value(t.id)) for t in orig_tasks[::2]],
-    )
-
-    ls.views.create(project=project.id, data=dict(title='Test View', filters=filters, ordering=['-' + Column.id]))
+def test_get_tasks_from_view(test_project_with_view):
+    ls, project, orig_tasks, view = test_project_with_view
     views = ls.views.list(project=project.id)
     assert len(views) == 1
-    view = views[0]
+    found_view = views[0]
+
+    assert found_view.id == view.id
     tasks = []
     for task in ls.tasks.list(view=view.id):
         tasks.append(task)