| """ |
| Basic tests for core components |
| """ |
|
|
| import pytest |
| from src.graph_rag_service.core.models import Entity, Relationship, Chunk |
|
|
|
|
| def test_entity_creation(): |
| """Test creating an entity""" |
| entity = Entity( |
| name="Acme Corp", |
| type="Company", |
| properties={"industry": "Technology", "founded": "2020"} |
| ) |
| |
| assert entity.name == "Acme Corp" |
| assert entity.type == "Company" |
| assert entity.properties["industry"] == "Technology" |
| assert entity.confidence == 1.0 |
|
|
|
|
| def test_relationship_creation(): |
| """Test creating a relationship""" |
| rel = Relationship( |
| source="Acme Corp", |
| target="Partner Inc", |
| type="PARTNERS_WITH", |
| confidence=0.9 |
| ) |
| |
| assert rel.source == "Acme Corp" |
| assert rel.target == "Partner Inc" |
| assert rel.type == "PARTNERS_WITH" |
| assert rel.confidence == 0.9 |
|
|
|
|
| def test_chunk_creation(): |
| """Test creating a chunk""" |
| chunk = Chunk( |
| text="Sample text content", |
| document_id="doc123", |
| chunk_index=0 |
| ) |
| |
| assert chunk.text == "Sample text content" |
| assert chunk.document_id == "doc123" |
| assert chunk.chunk_index == 0 |
|
|
|
|
| @pytest.mark.asyncio |
| async def test_llm_factory(): |
| """Test LLM factory creation""" |
| from src.graph_rag_service.core.llm_factory import LLMFactory |
| |
| |
| provider = LLMFactory.create() |
| assert provider is not None |
| assert provider.provider_name in ["ollama", "openai", "anthropic", "gemini"] |
|
|
|
|
| def test_config_loading(): |
| """Test configuration loading""" |
| from src.graph_rag_service.config import settings |
| |
| assert settings.app_name == "Graph RAG Service" |
| assert settings.app_version == "0.1.0" |
| assert settings.default_llm_provider in ["ollama", "openai", "anthropic", "gemini"] |
| assert settings.chunk_size > 0 |
| assert settings.default_top_k > 0 |
|
|
|
|
| if __name__ == "__main__": |
| pytest.main([__file__, "-v"]) |
|
|
