Skip to content

Commit 700fefe

Browse files
author
Val Brodsky
committed
Fix test_user_and_org.py
1 parent ba2990d commit 700fefe

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed
Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
from labelbox.schema.project import Project
2+
3+
14
def test_user(client):
25
user = client.get_user()
36
assert user.uid is not None
@@ -10,14 +13,11 @@ def test_organization(client):
1013
assert client.get_user() in set(organization.users())
1114

1215

13-
def test_user_and_org_projects(project):
14-
client = project.client
16+
def test_user_and_org_projects(client, project):
1517
user = client.get_user()
1618
org = client.get_organization()
17-
user_projects = set(user.projects())
18-
org_projects = set(org.projects())
19+
user_project = user.projects(where=Project.uid == project.uid)
20+
org_project = org.projects(where=Project.uid == project.uid)
1921

20-
assert project.created_by() == user
21-
assert project.organization() == org
22-
assert project in user_projects
23-
assert project in org_projects
22+
assert user_project
23+
assert org_project

0 commit comments

Comments
 (0)