From 18bc8afeb73855764d8a0b4ebb432dc5e0c76793 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Fri, 21 Sep 2018 13:55:49 -0400 Subject: [PATCH] Formatting --- atst/models/workspace_role.py | 8 ++++++-- tests/domain/test_workspace_users.py | 10 ++++++++-- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/atst/models/workspace_role.py b/atst/models/workspace_role.py index bcc2279b..da6b0ea4 100644 --- a/atst/models/workspace_role.py +++ b/atst/models/workspace_role.py @@ -10,13 +10,17 @@ class WorkspaceRole(Base, mixins.TimestampsMixin): __tablename__ = "workspace_roles" id = Id() - workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True, nullable=False) + workspace_id = Column( + UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True, nullable=False + ) workspace = relationship("Workspace", back_populates="roles") role_id = Column(UUID(as_uuid=True), ForeignKey("roles.id"), nullable=False) role = relationship("Role") - user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), index=True, nullable=False) + user_id = Column( + UUID(as_uuid=True), ForeignKey("users.id"), index=True, nullable=False + ) Index( diff --git a/tests/domain/test_workspace_users.py b/tests/domain/test_workspace_users.py index c07473f2..6e3aa710 100644 --- a/tests/domain/test_workspace_users.py +++ b/tests/domain/test_workspace_users.py @@ -12,7 +12,10 @@ def test_can_create_new_workspace_user(): assert workspace_users[0].user.id == new_user.id assert workspace_users[0].user.atat_role.name == new_user.atat_role.name - assert workspace_users[0].workspace_role.role.name == new_user.workspace_roles[0].role.name + assert ( + workspace_users[0].workspace_role.role.name + == new_user.workspace_roles[0].role.name + ) def test_can_update_existing_workspace_user(): @@ -27,4 +30,7 @@ def test_can_update_existing_workspace_user(): ) assert workspace_users[0].user.atat_role.name == new_user.atat_role.name - assert workspace_users[0].workspace_role.role.name == new_user.workspace_roles[0].role.name + assert ( + workspace_users[0].workspace_role.role.name + == new_user.workspace_roles[0].role.name + )