fix(tests): resolve asyncio event loop issue in translation integration tests
This commit is contained in:
parent
d70957978e
commit
aac115cc84
2 changed files with 22 additions and 0 deletions
3
.gitignore
vendored
3
.gitignore
vendored
|
|
@ -6,6 +6,9 @@ cognee/.data/
|
|||
|
||||
code_pipeline_output*/
|
||||
|
||||
# Test output files
|
||||
test_outputs/
|
||||
|
||||
*.lance/
|
||||
.DS_Store
|
||||
# Byte-compiled / optimized / DLL files
|
||||
|
|
|
|||
|
|
@ -18,10 +18,24 @@ def has_llm_api_key():
|
|||
return bool(os.environ.get("LLM_API_KEY"))
|
||||
|
||||
|
||||
async def reset_engines():
|
||||
"""Reset cached engines to avoid event loop issues between tests."""
|
||||
from cognee.infrastructure.databases.graph.get_graph_engine import create_graph_engine
|
||||
from cognee.infrastructure.databases.relational.create_relational_engine import (
|
||||
create_relational_engine,
|
||||
)
|
||||
from cognee.infrastructure.databases.vector.create_vector_engine import create_vector_engine
|
||||
|
||||
create_graph_engine.cache_clear()
|
||||
create_vector_engine.cache_clear()
|
||||
create_relational_engine.cache_clear()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_quick_translation():
|
||||
"""Quick smoke test for translation feature"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
@ -43,6 +57,7 @@ async def test_quick_translation():
|
|||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_translation_basic():
|
||||
"""Test basic translation functionality with English text"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
@ -69,6 +84,7 @@ async def test_translation_basic():
|
|||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_translation_spanish():
|
||||
"""Test translation with Spanish text"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
@ -100,6 +116,7 @@ async def test_translation_spanish():
|
|||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_translation_french():
|
||||
"""Test translation with French text"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
@ -131,6 +148,7 @@ async def test_translation_french():
|
|||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_translation_disabled():
|
||||
"""Test that cognify works without translation"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
@ -149,6 +167,7 @@ async def test_translation_disabled():
|
|||
@pytest.mark.skipif(not has_llm_api_key(), reason="No LLM API key available")
|
||||
async def test_translation_mixed_languages():
|
||||
"""Test with multiple documents in different languages"""
|
||||
await reset_engines()
|
||||
await prune.prune_data()
|
||||
await prune.prune_system(metadata=True)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue