From 4e84588a944871be5ed23b679da0ebd45df44892 Mon Sep 17 00:00:00 2001
From: Vito Sansevero <vito.sansevero@gmail.com>
Date: Tue, 23 Sep 2025 07:39:27 -0700
Subject: [PATCH] style(tests): improve code formatting consistency

---
 tests/test_embedding_autocomplete_fixed.py | 46 +++++++++++++---------
 1 file changed, 28 insertions(+), 18 deletions(-)

diff --git a/tests/test_embedding_autocomplete_fixed.py b/tests/test_embedding_autocomplete_fixed.py
index 61df010..7d50f3a 100644
--- a/tests/test_embedding_autocomplete_fixed.py
+++ b/tests/test_embedding_autocomplete_fixed.py
@@ -28,7 +28,9 @@ class TestMemoryManagement:
         # Mock autocomplete instance
         autocomplete = Mock()
         autocomplete.activeWidgets = weakref.WeakSet()
-        autocomplete.widgetCleanupMap = weakref.WeakKeyDictionary()  # Python equivalent of WeakMap
+        autocomplete.widgetCleanupMap = (
+            weakref.WeakKeyDictionary()
+        )  # Python equivalent of WeakMap
 
         # Simulate attaching widget
         autocomplete.activeWidgets.add(widget)
@@ -60,10 +62,16 @@ class TestMemoryManagement:
         autocomplete.suggestionContainer = mock_container
 
         # Simulate cleanup
-        autocomplete.cleanup = Mock(side_effect=lambda: (
-            mock_container.parentNode.removeChild(mock_container) if mock_container.parentNode else None,
-            setattr(autocomplete, 'suggestionContainer', None)
-        ))
+        autocomplete.cleanup = Mock(
+            side_effect=lambda: (
+                (
+                    mock_container.parentNode.removeChild(mock_container)
+                    if mock_container.parentNode
+                    else None
+                ),
+                setattr(autocomplete, "suggestionContainer", None),
+            )
+        )
 
         autocomplete.cleanup()
 
@@ -92,10 +100,10 @@ class TestMemoryManagement:
 
         # Simulate attaching autocomplete
         handlers = {
-            'input': Mock(),
-            'keydown': Mock(),
-            'blur': Mock(),
-            'scroll': Mock()
+            "input": Mock(),
+            "keydown": Mock(),
+            "blur": Mock(),
+            "scroll": Mock(),
         }
 
         for event_type, handler in handlers.items():
@@ -108,7 +116,10 @@ class TestMemoryManagement:
         # All listeners should be removed
         assert textarea.removeEventListener.call_count == 4
         for event_type in handlers.keys():
-            assert any(call[0][0] == event_type for call in textarea.removeEventListener.call_args_list)
+            assert any(
+                call[0][0] == event_type
+                for call in textarea.removeEventListener.call_args_list
+            )
 
     def test_pending_fetch_cleanup(self):
         """Test that pending fetch requests are aborted on cleanup."""
@@ -227,8 +238,7 @@ class TestResourceFetching:
         threads = []
         for i in range(10):
             thread = threading.Thread(
-                target=update_embeddings,
-                args=([f"embedding_{i}"],)
+                target=update_embeddings, args=([f"embedding_{i}"],)
             )
             threads.append(thread)
             thread.start()
@@ -460,13 +470,13 @@ class TestIntegration:
         def attach_widgets(nodes):
             for node in nodes:
                 for widget in node.widgets:
-                    if hasattr(widget.inputEl, 'tagName'):
+                    if hasattr(widget.inputEl, "tagName"):
                         active_widgets.append(widget)
 
         def clear_graph():
             # Cleanup all widgets
             for widget in active_widgets:
-                if hasattr(widget, 'onRemoved') and widget.onRemoved:
+                if hasattr(widget, "onRemoved") and widget.onRemoved:
                     widget.onRemoved()
             active_widgets.clear()
 
@@ -489,17 +499,17 @@ class TestIntegration:
 
         def track_listener(event_type, handler):
             nonlocal cleanup_handler
-            if event_type == 'beforeunload':
+            if event_type == "beforeunload":
                 cleanup_handler = handler
 
         mock_window.addEventListener.side_effect = track_listener
 
         # Simulate autocomplete setup with window listener
-        mock_window.addEventListener('beforeunload', lambda: None)
+        mock_window.addEventListener("beforeunload", lambda: None)
 
         # Verify listener was added
         assert mock_window.addEventListener.called
-        assert mock_window.addEventListener.call_args[0][0] == 'beforeunload'
+        assert mock_window.addEventListener.call_args[0][0] == "beforeunload"
 
         # Simulate cleanup being called
         if cleanup_handler:
@@ -560,4 +570,4 @@ class TestPerformance:
 
 
 if __name__ == "__main__":
-    pytest.main([__file__, "-v"])
\ No newline at end of file
+    pytest.main([__file__, "-v"])