chore: linting

Author: cofin

tests/integration/test_inheritance_patterns.py

@@ -45,7 +45,9 @@ class Cat(Animal):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert data
         dog = Dog(name="Buddy", breed="Golden Retriever")
@@ -89,7 +91,9 @@ class Engineer(Employee):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert data
         manager = Manager(name="Alice", manager_level=5)
@@ -133,7 +137,9 @@ class Car(Vehicle):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert and update
         car = Car(brand="Toyota", doors=4)
@@ -168,7 +174,9 @@ class Book(Product):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert
         book = Book(name="Python Guide", author="John Doe")
@@ -210,7 +218,9 @@ class Student(Person):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert data
         student = Student(name="Alice", grade="A")
@@ -259,7 +269,9 @@ class ContentRepository(SQLAlchemyAsyncRepository[Content]):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Use repository
         repo = ContentRepository(session=async_session)
@@ -311,7 +323,9 @@ class Company(Organization):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert at different levels
         entity = Entity(name="General Entity")
@@ -355,7 +369,9 @@ class Report(Document):
         # Create tables
         async with async_session.begin():
             await async_session.connection(execution_options={"isolation_level": "AUTOCOMMIT"})
-            await async_session.run_sync(lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind()))
+            await async_session.run_sync(
+                lambda sync_session: base.UUIDBase.metadata.create_all(bind=sync_session.get_bind())
+            )
 
         # Insert
         report = Report(title="Q4 Report", department="Sales")

tests/unit/test_base_inheritance.py

@@ -9,7 +9,6 @@
 from typing import Optional
 from uuid import UUID
 
-import pytest
 from sqlalchemy import ForeignKey, String
 from sqlalchemy.orm import Mapped, declarative_mixin, mapped_column