improved tests - fixing the linting

This commit is contained in:
yuisheaven
2025-10-23 22:56:25 +02:00
parent 6730dd4a4b
commit db79afacb9
2 changed files with 8 additions and 9 deletions
@@ -1,10 +1,8 @@
"""Integration tests for Unstructured API functionality."""
import base64
import json
import logging
import uuid
from typing import Any, Dict
from io import BytesIO
import pytest
+8 -7
View File
@@ -1,13 +1,14 @@
"""Unit tests for Unstructured API configuration."""
import os
import pytest
from nextcloud_mcp_server.config import (
get_unstructured_strategy,
get_unstructured_languages,
)
from nextcloud_mcp_server.client.unstructured_client import UnstructuredClient
from nextcloud_mcp_server.config import (
get_unstructured_languages,
get_unstructured_strategy,
)
class TestUnstructuredStrategy:
@@ -134,7 +135,7 @@ class TestUnstructuredClientConfiguration:
os.environ["UNSTRUCTURED_API_URL"] = "http://test:8000"
try:
client = UnstructuredClient()
_client = UnstructuredClient()
# The partition_document method should use get_unstructured_strategy() when strategy is None
# We can't test the actual call without a running API, but we can verify the config is read
assert get_unstructured_strategy() == "auto"
@@ -148,7 +149,7 @@ class TestUnstructuredClientConfiguration:
os.environ["UNSTRUCTURED_API_URL"] = "http://test:8000"
try:
client = UnstructuredClient()
_client = UnstructuredClient()
# The partition_document method should use get_unstructured_languages() when languages is None
assert get_unstructured_languages() == ["eng", "deu"]
finally:
@@ -162,7 +163,7 @@ class TestUnstructuredClientConfiguration:
os.environ["UNSTRUCTURED_API_URL"] = "http://test:8000"
try:
client = UnstructuredClient()
_client = UnstructuredClient()
assert get_unstructured_strategy() == "hi_res"
assert get_unstructured_languages() == ["eng", "fra", "spa"]
finally: