diff --git a/argilla/tests/integration/test_manage_users.py b/argilla/tests/integration/test_manage_users.py index 4c902fca45..fdbf6a7860 100644 --- a/argilla/tests/integration/test_manage_users.py +++ b/argilla/tests/integration/test_manage_users.py @@ -27,10 +27,11 @@ def test_create_user(self, client: Argilla): assert client.users(username=user.username).id == user.id def test_create_user_with_id(self, client: Argilla): - user = User(id=uuid.uuid4(), username=f"test_user_{uuid.uuid4()}", password="test_password") + user_id = uuid.uuid4() + user = User(id=user_id, username=f"test_user_{uuid.uuid4()}", password="test_password") client.users.add(user) assert user.id is not None - assert client.users(username=user.username).id == user.id + assert client.users(username=user.username).id == user_id def test_create_user_without_password(self, client: Argilla): user = User(username=f"test_user_{uuid.uuid4()}") diff --git a/argilla/tests/integration/test_manage_workspaces.py b/argilla/tests/integration/test_manage_workspaces.py index 3a93a59aec..6c5167679e 100644 --- a/argilla/tests/integration/test_manage_workspaces.py +++ b/argilla/tests/integration/test_manage_workspaces.py @@ -25,11 +25,11 @@ def test_create_workspace(self, client: Argilla): assert client.api.workspaces.exists(workspace.id) def test_create_workspace_with_id(self, client: Argilla): - workspace = Workspace(id=uuid.uuid4(), name=f"test_workspace{uuid.uuid4()}") + workspace_id = uuid.uuid4() + workspace = Workspace(id=workspace_id, name=f"test_workspace{uuid.uuid4()}") client.workspaces.add(workspace) - assert workspace in client.workspaces - assert client.workspaces(workspace.name).id == workspace.id + assert client.workspaces(workspace.name).id == workspace_id def test_create_and_delete_workspace(self, client: Argilla): workspace = client.workspaces(name="test_workspace")