fix(040): fix ruff errors in test files — missing ParquetStore import, write_staged_batch

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: satra

library/tests/test_pipeline_e2e.py

@@ -12,7 +12,8 @@
 from undata_library.commit import commit_staged
 from undata_library.enrich import enrich_elements
 from undata_library.ingest import ingest_source
-from undata_library.staging import create_staging_dir, generate_run_id
+from undata_library.staging import create_staging_dir, generate_run_id, write_staged_batch
+from undata_library.storage.parquet_store import ParquetStore
 from undata_library.utils import write_yaml
 
 
@@ -71,12 +72,18 @@ def test_synthetic_element_committed(self, tmp_path):
         # Add a synthetic element to a new staging run
         run_id = generate_run_id()
         staging = create_staging_dir(tmp_path, run_id)
-        write_yaml(
-            staging / "elements" / "synthetic_test.yaml",
-            {
-                "semantic": {"data_type": "float", "unit": "meter"},
-                "provenance": [{"source": "test", "class": "synthetic", "name": "test_distance"}],
-            },
+        write_staged_batch(
+            staging,
+            "elements",
+            [
+                {
+                    "semantic": {"data_type": "float", "unit": "meter"},
+                    "provenance": [
+                        {"source": "test", "class": "synthetic", "name": "test_distance"}
+                    ],
+                }
+            ],
+            source="test",
         )
         commit_staged(staging, tmp_path)
 

library/tests/test_staging.py

@@ -46,7 +46,12 @@ def test_count_staged(tmp_path):
     write_staged_batch(
         staging,
         "elements",
-        [{"semantic": {"data_type": "string"}, "provenance": [{"source": "t", "class": "X", "name": "a"}]}],
+        [
+            {
+                "semantic": {"data_type": "string"},
+                "provenance": [{"source": "t", "class": "X", "name": "a"}],
+            }
+        ],
         source="test",
     )
     assert count_staged(staging, "elements") == 1
@@ -57,7 +62,12 @@ def test_iter_staged(tmp_path):
     write_staged_batch(
         staging,
         "elements",
-        [{"semantic": {"data_type": "string"}, "provenance": [{"source": "t", "class": "X", "name": "a"}]}],
+        [
+            {
+                "semantic": {"data_type": "string"},
+                "provenance": [{"source": "t", "class": "X", "name": "a"}],
+            }
+        ],
         source="test",
     )
     entities = list(iter_staged(staging, "elements"))