|
1 | 1 | from typing import List, Optional |
2 | 2 |
|
3 | 3 | import pytest |
4 | | -from sqlalchemy.exc import IntegrityError |
5 | | -from sqlalchemy.orm import RelationshipProperty |
6 | | -from sqlmodel import Field, Relationship, Session, SQLModel, create_engine, select |
7 | | -from typing_extensions import Annotated |
| 4 | +from sqlalchemy.exc import IntegrityError |
| 5 | +from sqlalchemy.orm import RelationshipProperty |
| 6 | +from sqlmodel import Field, Relationship, Session, SQLModel, create_engine, select |
| 7 | +from typing_extensions import Annotated |
8 | 8 |
|
9 | 9 | from tests.conftest import needs_pydanticv2 |
10 | 10 |
|
@@ -139,8 +139,8 @@ class UserPermission(SQLModel, table=True): |
139 | 139 | engine = create_engine("sqlite://") |
140 | 140 | SQLModel.metadata.create_all(engine) |
141 | 141 |
|
142 | | - pk_column_names = {column.name for column in UserPermission.__table__.primary_key} |
143 | | - assert pk_column_names == {"user_id", "resource_id"} |
| 142 | + pk_column_names = {column.name for column in UserPermission.__table__.primary_key} |
| 143 | + assert pk_column_names == {"user_id", "resource_id"} |
144 | 144 |
|
145 | 145 | with Session(engine) as session: |
146 | 146 | perm1 = UserPermission(user_id=1, resource_id=1, permission="read") |
@@ -175,8 +175,8 @@ class UserPermission(SQLModel, table=True): |
175 | 175 | engine = create_engine("sqlite://") |
176 | 176 | SQLModel.metadata.create_all(engine) |
177 | 177 |
|
178 | | - pk_column_names = {column.name for column in UserPermission.__table__.primary_key} |
179 | | - assert pk_column_names == {"user_id", "resource_id"} |
| 178 | + pk_column_names = {column.name for column in UserPermission.__table__.primary_key} |
| 179 | + assert pk_column_names == {"user_id", "resource_id"} |
180 | 180 |
|
181 | 181 | with Session(engine) as session: |
182 | 182 | perm1 = UserPermission(user_id=1, resource_id=1, permission="read") |
|
0 commit comments