Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions tests/integration/annotation_import/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,8 @@ def ontology():

@pytest.fixture
def configured_project(client, ontology, rand_gen, image_url):
project = client.create_project(name=rand_gen(str))
project = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)
dataset = client.create_dataset(name=rand_gen(str))
editor = list(
client.get_labeling_frontends(
Expand All @@ -143,7 +144,8 @@ def configured_project(client, ontology, rand_gen, image_url):

@pytest.fixture
def configured_project_pdf(client, ontology, rand_gen, pdf_url):
project = client.create_project(name=rand_gen(str))
project = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)
dataset = client.create_dataset(name=rand_gen(str))
editor = list(
client.get_labeling_frontends(
Expand Down
6 changes: 4 additions & 2 deletions tests/integration/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,8 @@ def pdf_url(client):

@pytest.fixture
def project(client, rand_gen):
project = client.create_project(name=rand_gen(str))
project = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)
yield project
project.delete()

Expand Down Expand Up @@ -346,7 +347,8 @@ def create_label():

@pytest.fixture
def configured_project_with_complex_ontology(client, rand_gen, image_url):
project = client.create_project(name=rand_gen(str))
project = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)
dataset = client.create_dataset(name=rand_gen(str), projects=project)
data_row = dataset.create_data_row(row_data=image_url)
editor = list(
Expand Down
7 changes: 4 additions & 3 deletions tests/integration/test_filtering.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,15 @@

from labelbox import Project
from labelbox.exceptions import InvalidQueryError
from labelbox.schema.queue_mode import QueueMode


# Avoid assertions using equality to prevent intermittent failures due to
# other builds simultaneously adding projects to test org
def test_where(client):
p_a = client.create_project(name="a")
p_b = client.create_project(name="b")
p_c = client.create_project(name="c")
p_a = client.create_project(name="a", queue_mode=QueueMode.Dataset)
p_b = client.create_project(name="b", queue_mode=QueueMode.Dataset)
p_c = client.create_project(name="c", queue_mode=QueueMode.Dataset)

def _get(f, where=None):
date_where = Project.created_at >= p_a.created_at
Expand Down
7 changes: 5 additions & 2 deletions tests/integration/test_relationships.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
import pytest

from labelbox.exceptions import InvalidQueryError
from labelbox.schema.queue_mode import QueueMode


def test_project_dataset(client, rand_gen):
project = client.create_project(name=rand_gen(str))
project = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)
dataset = client.create_dataset(name=rand_gen(str))

assert len(list(project.datasets())) == 0
Expand All @@ -17,7 +19,8 @@ def test_project_dataset(client, rand_gen):
assert {ds.uid for ds in project.datasets()} == {dataset.uid}
assert {pr.uid for pr in dataset.projects()} == {project.uid}

project_2 = client.create_project(name=rand_gen(str))
project_2 = client.create_project(name=rand_gen(str),
queue_mode=QueueMode.Dataset)

# Currently it's not possible to connect a project and dataset
# by updating dataset.
Expand Down
13 changes: 10 additions & 3 deletions tests/integration/test_sorting.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,21 @@
import pytest

from labelbox import Project
from labelbox.schema.queue_mode import QueueMode


@pytest.mark.xfail(reason="Relationship sorting not implemented correctly "
"on the server-side")
def test_relationship_sorting(client):
a = client.create_project(name="a", description="b")
b = client.create_project(name="b", description="c")
c = client.create_project(name="c", description="a")
a = client.create_project(name="a",
description="b",
queue_mode=QueueMode.Dataset)
b = client.create_project(name="b",
description="c",
queue_mode=QueueMode.Dataset)
c = client.create_project(name="c",
description="a",
queue_mode=QueueMode.Dataset)

dataset = client.create_dataset(name="Dataset")
a.datasets.connect(dataset)
Expand Down