style: Format test file to fix Ruff format check

Co-Authored-By: AJ Steers <[email protected]>

Author: devin-ai-integration[bot]

unit_tests/sources/declarative/schema/test_inferred_schema_loader.py

@@ -80,11 +80,13 @@
 def mock_retriever():
     """Create a mock retriever that returns sample records."""
     retriever = MagicMock()
-    retriever.read_records.return_value = iter([
-        {"id": 1, "name": "Alice", "age": 30, "active": True},
-        {"id": 2, "name": "Bob", "age": 25, "active": False},
-        {"id": 3, "name": "Charlie", "age": 35, "active": True},
-    ])
+    retriever.read_records.return_value = iter(
+        [
+            {"id": 1, "name": "Alice", "age": 30, "active": True},
+            {"id": 2, "name": "Bob", "age": 25, "active": False},
+            {"id": 3, "name": "Charlie", "age": 35, "active": True},
+        ]
+    )
     return retriever
 
 
@@ -109,12 +111,12 @@ def test_inferred_schema_loader_basic(inferred_schema_loader):
     assert "$schema" in schema
     assert schema["type"] == "object"
     assert "properties" in schema
-    
+
     assert "id" in schema["properties"]
     assert "name" in schema["properties"]
     assert "age" in schema["properties"]
     assert "active" in schema["properties"]
-    
+
     assert "number" in schema["properties"]["id"]["type"]
     assert "string" in schema["properties"]["name"]["type"]
     assert "number" in schema["properties"]["age"]["type"]
@@ -125,7 +127,7 @@ def test_inferred_schema_loader_empty_records():
     """Test that InferredSchemaLoader returns empty schema when no records are available."""
     retriever = MagicMock()
     retriever.read_records.return_value = iter([])
-    
+
     config = MagicMock()
     parameters = {"name": "users"}
     loader = InferredSchemaLoader(
@@ -135,21 +137,18 @@ def test_inferred_schema_loader_empty_records():
         record_sample_size=100,
         stream_name="users",
     )
-    
+
     schema = loader.get_json_schema()
-    
+
     assert schema == {}
 
 
 def test_inferred_schema_loader_respects_sample_size():
     """Test that InferredSchemaLoader respects the record_sample_size parameter."""
     retriever = MagicMock()
-    records = [
-        {"id": i, "name": f"User{i}"}
-        for i in range(10)
-    ]
+    records = [{"id": i, "name": f"User{i}"} for i in range(10)]
     retriever.read_records.return_value = iter(records)
-    
+
     config = MagicMock()
     parameters = {"name": "users"}
     loader = InferredSchemaLoader(
@@ -159,9 +158,9 @@ def test_inferred_schema_loader_respects_sample_size():
         record_sample_size=5,
         stream_name="users",
     )
-    
+
     schema = loader.get_json_schema()
-    
+
     assert "properties" in schema
     assert "id" in schema["properties"]
     assert "name" in schema["properties"]
@@ -171,7 +170,7 @@ def test_inferred_schema_loader_handles_errors():
     """Test that InferredSchemaLoader handles errors gracefully."""
     retriever = MagicMock()
     retriever.read_records.side_effect = Exception("API Error")
-    
+
     config = MagicMock()
     parameters = {"name": "users"}
     loader = InferredSchemaLoader(
@@ -181,36 +180,30 @@ def test_inferred_schema_loader_handles_errors():
         record_sample_size=100,
         stream_name="users",
     )
-    
+
     schema = loader.get_json_schema()
-    
+
     assert schema == {}
 
 
 def test_inferred_schema_loader_with_nested_objects():
     """Test that InferredSchemaLoader handles nested objects correctly."""
     retriever = MagicMock()
-    retriever.read_records.return_value = iter([
-        {
-            "id": 1,
-            "name": "Alice",
-            "address": {
-                "street": "123 Main St",
-                "city": "Springfield",
-                "zip": "12345"
-            }
-        },
-        {
-            "id": 2,
-            "name": "Bob",
-            "address": {
-                "street": "456 Oak Ave",
-                "city": "Shelbyville",
-                "zip": "67890"
-            }
-        },
-    ])
-    
+    retriever.read_records.return_value = iter(
+        [
+            {
+                "id": 1,
+                "name": "Alice",
+                "address": {"street": "123 Main St", "city": "Springfield", "zip": "12345"},
+            },
+            {
+                "id": 2,
+                "name": "Bob",
+                "address": {"street": "456 Oak Ave", "city": "Shelbyville", "zip": "67890"},
+            },
+        ]
+    )
+
     config = MagicMock()
     parameters = {"name": "users"}
     loader = InferredSchemaLoader(
@@ -220,9 +213,9 @@ def test_inferred_schema_loader_with_nested_objects():
         record_sample_size=2,
         stream_name="users",
     )
-    
+
     schema = loader.get_json_schema()
-    
+
     assert "properties" in schema
     assert "address" in schema["properties"]
     assert "object" in schema["properties"]["address"]["type"]
@@ -231,11 +224,13 @@ def test_inferred_schema_loader_with_nested_objects():
 def test_inferred_schema_loader_with_arrays():
     """Test that InferredSchemaLoader handles arrays correctly."""
     retriever = MagicMock()
-    retriever.read_records.return_value = iter([
-        {"id": 1, "name": "Alice", "tags": ["admin", "user"]},
-        {"id": 2, "name": "Bob", "tags": ["user", "guest"]},
-    ])
-    
+    retriever.read_records.return_value = iter(
+        [
+            {"id": 1, "name": "Alice", "tags": ["admin", "user"]},
+            {"id": 2, "name": "Bob", "tags": ["user", "guest"]},
+        ]
+    )
+
     config = MagicMock()
     parameters = {"name": "users"}
     loader = InferredSchemaLoader(
@@ -245,9 +240,9 @@ def test_inferred_schema_loader_with_arrays():
         record_sample_size=2,
         stream_name="users",
     )
-    
+
     schema = loader.get_json_schema()
-    
+
     assert "properties" in schema
     assert "tags" in schema["properties"]
     assert "array" in schema["properties"]["tags"]["type"]