Nomad changes

e2e/python/tests/test_plugin_api.py

@@ -0,0 +1,176 @@
+# This file is auto-generated by alef — DO NOT EDIT.
+# alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
+# To regenerate: alef generate
+# To verify freshness: alef verify --exit-code
+# Issues & docs: https://github.com/kreuzberg-dev/alef
+
+"""E2e tests for category: plugin_api."""
+
+import pytest  # noqa: F401
+from kreuzberg import register_document_extractor, register_embedding_backend, register_ocr_backend, register_post_processor, register_renderer, register_validator, unregister_document_extractor, unregister_embedding_backend, unregister_post_processor, unregister_renderer, unregister_validator, unregister_ocr_backend
+
+
+def _alef_e2e_text(value: object) -> str:
+    return "" if value is None else str(value)
+
+
+def _alef_e2e_item_texts(item: object) -> tuple[str, ...]:
+    raw_items = getattr(item, "items", None)
+    items_text = " ".join(str(value) for value in raw_items) if isinstance(raw_items, list) else ""
+    return (
+        _alef_e2e_text(item),
+        _alef_e2e_text(getattr(item, "kind", None)),
+        _alef_e2e_text(getattr(item, "name", None)),
+        _alef_e2e_text(getattr(item, "source", None)),
+        _alef_e2e_text(getattr(item, "alias", None)),
+        _alef_e2e_text(getattr(item, "text", None)),
+        _alef_e2e_text(getattr(item, "signature", None)),
+        items_text,
+    )
+
+
+def test_register_document_extractor_trait_bridge() -> None:
+    """register_document_extractor: trait bridge."""
+    class _TestStub_register_document_extractor_trait_bridge:
+        def name(self):
+            return "test-extractor"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        async def extract_bytes(self, _p0, _p1, _p2):
+            return {}
+        def supported_mime_types(self):
+            return []
+
+
+    _ = register_document_extractor(_TestStub_register_document_extractor_trait_bridge())
+    unregister_document_extractor("test-extractor")
+
+
+def test_register_embedding_backend_trait_bridge() -> None:
+    """register_embedding_backend: trait bridge."""
+    class _TestStub_register_embedding_backend_trait_bridge:
+        def name(self):
+            return "test-embedding-backend"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        def dimensions(self):
+            return 1
+        async def embed(self, _p0):
+            return []
+
+
+    _ = register_embedding_backend(_TestStub_register_embedding_backend_trait_bridge())
+    unregister_embedding_backend("test-embedding-backend")
+
+
+def test_register_ocr_backend_trait_bridge() -> None:
+    """register_ocr_backend: trait bridge."""
+    class _TestStub_register_ocr_backend_trait_bridge:
+        def name(self):
+            return "test-backend"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        async def process_image(self, _p0, _p1):
+            return {}
+        def supports_language(self, _p0):
+            return False
+        def backend_type(self):
+            return {}
+
+
+    _ = register_ocr_backend(_TestStub_register_ocr_backend_trait_bridge())
+    unregister_ocr_backend("test-backend")
+
+
+def test_register_post_processor_trait_bridge() -> None:
+    """register_post_processor: trait bridge."""
+    class _TestStub_register_post_processor_trait_bridge:
+        def name(self):
+            return "test-processor"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        async def process(self, _p0, _p1):
+            return None
+        def processing_stage(self):
+            return {}
+
+
+    _ = register_post_processor(_TestStub_register_post_processor_trait_bridge())
+    unregister_post_processor("test-processor")
+
+
+def test_register_renderer_trait_bridge() -> None:
+    """register_renderer: trait bridge."""
+    class _TestStub_register_renderer_trait_bridge:
+        def name(self):
+            return "test-renderer"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        def render(self, _p0):
+            return ""
+
+
+    _ = register_renderer(_TestStub_register_renderer_trait_bridge())
+    unregister_renderer("test-renderer")
+
+
+def test_register_validator_trait_bridge() -> None:
+    """register_validator: trait bridge."""
+    class _TestStub_register_validator_trait_bridge:
+        def name(self):
+            return "test-validator"
+        def initialize(self):
+            pass
+        def shutdown(self):
+            pass
+        async def validate(self, _p0, _p1):
+            return None
+
+
+    _ = register_validator(_TestStub_register_validator_trait_bridge())
+    unregister_validator("test-validator")
+
+
+def test_unregister_document_extractor_after_register() -> None:
+    """unregister_document_extractor."""
+    name = "test-extractor"
+
+    _ = unregister_document_extractor(name)
+
+
+def test_unregister_embedding_backend_after_register() -> None:
+    """unregister_embedding_backend."""
+    name = "test-embedding-backend"
+
+    _ = unregister_embedding_backend(name)
+
+
+def test_unregister_post_processor_after_register() -> None:
+    """unregister_post_processor."""
+    name = "test-processor"
+
+    _ = unregister_post_processor(name)
+
+
+def test_unregister_renderer_after_register() -> None:
+    """unregister_renderer."""
+    name = "test-renderer"
+
+    _ = unregister_renderer(name)
+
+
+def test_unregister_validator_after_register() -> None:
+    """unregister_validator."""
+    name = "test-validator"
+
+    _ = unregister_validator(name)