ruff format

Author: wey-gu

examples/conftest.py

@@ -39,19 +39,21 @@
 @pytest.fixture(scope="module")
 def pglite_manager() -> Generator[PGliteManager, None, None]:
     """Isolated PGlite manager for examples - one per test module.
-    
+
     This overrides the session-scoped fixture from the main package
     to provide better isolation when running all tests together.
     """
     # Create unique configuration to prevent socket conflicts
     config = PGliteConfig()
-    
+
     # Create a unique socket directory for this example module
     # PGlite expects socket_path to be the full path including .s.PGSQL.5432
-    socket_dir = Path(tempfile.gettempdir()) / f"py-pglite-example-{uuid.uuid4().hex[:8]}"
+    socket_dir = (
+        Path(tempfile.gettempdir()) / f"py-pglite-example-{uuid.uuid4().hex[:8]}"
+    )
     socket_dir.mkdir(mode=0o700, exist_ok=True)  # Restrict to user only
     config.socket_path = str(socket_dir / ".s.PGSQL.5432")
-    
+
     manager = PGliteManager(config)
     manager.start()
     manager.wait_for_ready()
@@ -94,28 +96,32 @@ def pglite_session(pglite_engine: Engine) -> Generator[Any, None, None]:
                 if table_names:
                     # Disable foreign key checks for faster cleanup
                     conn.execute(text("SET session_replication_role = replica;"))
-                    
+
                     # Truncate all tables
                     for table_name in table_names:
                         logger.info(f"Truncating table: {table_name}")
                         conn.execute(
-                            text(f'TRUNCATE TABLE "{table_name}" RESTART IDENTITY CASCADE;')
+                            text(
+                                f'TRUNCATE TABLE "{table_name}" RESTART IDENTITY CASCADE;'
+                            )
                         )
 
                     # Re-enable foreign key checks
                     conn.execute(text("SET session_replication_role = DEFAULT;"))
-                    
+
                     # Commit the cleanup
                     conn.commit()
                     logger.info("Database cleanup completed successfully")
                 else:
                     logger.info("No tables found to clean")
                 break  # Success, exit retry loop
-                
+
         except Exception as e:
             logger.info(f"Database cleanup attempt {attempt + 1} failed: {e}")
             if attempt == retry_count - 1:
-                logger.warning("Database cleanup failed after all retries, continuing anyway")
+                logger.warning(
+                    "Database cleanup failed after all retries, continuing anyway"
+                )
             else:
                 time.sleep(0.5)  # Brief pause before retry
 
@@ -144,4 +150,4 @@ def pglite_session(pglite_engine: Engine) -> Generator[Any, None, None]:
         try:
             session.close()
         except Exception as e:
-            logger.warning(f"Error closing session: {e}") 
+            logger.warning(f"Error closing session: {e}")

examples/quickstart/simple_performance.py

@@ -191,9 +191,7 @@ def measure_feature_power():
                 f"      → {result[1]} events/minute, avg score: {result[2]:.1f}",
             )
 
-            total_feature_time = (
-                json_time + array_time + window_time + time_series_time
-            )
+            total_feature_time = json_time + array_time + window_time + time_series_time
 
             print(f"\n   🚀 Total advanced features: {total_feature_time:.4f}s")
             print("   💥 SQLite equivalent: IMPOSSIBLE ❌")
@@ -263,12 +261,10 @@ def measure_raw_performance():
         sqlite_insert = time.time() - start
 
         print(
-            f"   py-pglite: {pglite_insert:.3f}s "
-            f"({1000 / pglite_insert:,.0f} rec/sec)"
+            f"   py-pglite: {pglite_insert:.3f}s ({1000 / pglite_insert:,.0f} rec/sec)"
         )
         print(
-            f"   SQLite:    {sqlite_insert:.3f}s "
-            f"({1000 / sqlite_insert:,.0f} rec/sec)"
+            f"   SQLite:    {sqlite_insert:.3f}s ({1000 / sqlite_insert:,.0f} rec/sec)"
         )
 
         if sqlite_insert < pglite_insert:
@@ -322,8 +318,7 @@ def generate_final_report(boot_times, feature_time, perf_times):
     print("\n" + "🎯 THE HONEST TRUTH" + "\n" + "=" * 60)
 
     print("📊 PERFORMANCE COMPARISON:")
-    print(f"   Boot Time:    SQLite {sqlite_boot:.3f}s vs "
-          f"py-pglite {pglite_boot:.2f}s")
+    print(f"   Boot Time:    SQLite {sqlite_boot:.3f}s vs py-pglite {pglite_boot:.2f}s")
     print(
         f"   Insert Speed: SQLite {1000 / sqlite_insert:,.0f}/s vs "
         f"py-pglite {1000 / pglite_insert:,.0f}/s"

examples/test_basic.py

@@ -5,7 +5,7 @@
 
 # Example model
 class BasicUser(SQLModel, table=True):
-    __table_args__ = {'extend_existing': True}
+    __table_args__ = {"extend_existing": True}
     id: int | None = Field(default=None, primary_key=True)
     name: str
     email: str
@@ -60,7 +60,9 @@ def test_user_update(pglite_session: Session):
     pglite_session.commit()
 
     # Verify the update
-    updated_user = pglite_session.exec(select(BasicUser).where(BasicUser.name == "David")).first()
+    updated_user = pglite_session.exec(
+        select(BasicUser).where(BasicUser.name == "David")
+    ).first()
     assert updated_user is not None
     assert updated_user.email == "david@new.com"
 

examples/test_utils.py

@@ -42,7 +42,7 @@ def test_database_cleanup_utils(pglite_engine):
             title="Python Testing Guide",
             author_id=author.id,
             isbn="978-0123456789",
-            published_year=2024
+            published_year=2024,
         )
         session.add(book)
         session.commit()
@@ -138,7 +138,7 @@ def test_partial_cleanup(pglite_engine):
             title="Temporary Book",
             author_id=author.id,
             isbn="978-0987654321",
-            published_year=2024
+            published_year=2024,
         )
         session.add(book)
         session.commit()
@@ -153,7 +153,9 @@ def test_partial_cleanup(pglite_engine):
 
     # Verify with exclude list in verification
     assert utils.verify_database_empty(pglite_engine, exclude_tables=["author"])
-    assert not utils.verify_database_empty(pglite_engine)  # Should be False due to author
+    assert not utils.verify_database_empty(
+        pglite_engine
+    )  # Should be False due to author
 
 
 def test_schema_operations(pglite_engine):
@@ -167,8 +169,10 @@ def test_schema_operations(pglite_engine):
     with Session(pglite_engine) as session:
         with session.connection() as conn:
             result = conn.execute(
-                text("SELECT schema_name FROM information_schema.schemata WHERE schema_name = :name"),
-                {"name": test_schema}
+                text(
+                    "SELECT schema_name FROM information_schema.schemata WHERE schema_name = :name"
+                ),
+                {"name": test_schema},
             )
             schemas = result.fetchall()
             assert len(schemas) == 1
@@ -180,8 +184,10 @@ def test_schema_operations(pglite_engine):
     with Session(pglite_engine) as session:
         with session.connection() as conn:
             result = conn.execute(
-                text("SELECT schema_name FROM information_schema.schemata WHERE schema_name = :name"),
-                {"name": test_schema}
+                text(
+                    "SELECT schema_name FROM information_schema.schemata WHERE schema_name = :name"
+                ),
+                {"name": test_schema},
             )
             schemas = result.fetchall()
             assert len(schemas) == 0
@@ -198,8 +204,8 @@ def test_combined_cleanup_fixture(pglite_session: Session):
         # Clean all tables manually to ensure fresh state
         conn.execute(text('DELETE FROM "book"'))
         conn.execute(text('DELETE FROM "author"'))
-        conn.execute(text('ALTER SEQUENCE author_id_seq RESTART WITH 1'))
-        conn.execute(text('ALTER SEQUENCE book_id_seq RESTART WITH 1'))
+        conn.execute(text("ALTER SEQUENCE author_id_seq RESTART WITH 1"))
+        conn.execute(text("ALTER SEQUENCE book_id_seq RESTART WITH 1"))
         pglite_session.commit()
 
     # Add test data directly using the existing session
@@ -213,7 +219,7 @@ def test_combined_cleanup_fixture(pglite_session: Session):
         title="Test Book",
         author_id=author.id,
         isbn="978-0111111111",
-        published_year=2024
+        published_year=2024,
     )
     pglite_session.add(book)
     pglite_session.commit()