|
|
""" |
|
|
Basic tests for GraphLLM components |
|
|
""" |
|
|
import pytest |
|
|
from models import Chunk, ChunkType, GraphNode, GraphEdge, Triple, NodeType, RelationType |
|
|
from config import settings |
|
|
|
|
|
|
|
|
def test_chunk_creation(): |
|
|
"""Test chunk model creation""" |
|
|
chunk = Chunk( |
|
|
pdf_id="test-pdf", |
|
|
page_number=1, |
|
|
char_range=(0, 100), |
|
|
type=ChunkType.PARAGRAPH, |
|
|
text="This is a test chunk." |
|
|
) |
|
|
|
|
|
assert chunk.pdf_id == "test-pdf" |
|
|
assert chunk.page_number == 1 |
|
|
assert chunk.type == ChunkType.PARAGRAPH |
|
|
assert chunk.text == "This is a test chunk." |
|
|
|
|
|
|
|
|
def test_graph_node_creation(): |
|
|
"""Test graph node creation""" |
|
|
node = GraphNode( |
|
|
label="Test Concept", |
|
|
type=NodeType.CONCEPT, |
|
|
aliases=["test", "concept"], |
|
|
supporting_chunks=[], |
|
|
importance_score=0.75 |
|
|
) |
|
|
|
|
|
assert node.label == "Test Concept" |
|
|
assert node.type == NodeType.CONCEPT |
|
|
assert node.importance_score == 0.75 |
|
|
|
|
|
|
|
|
def test_graph_edge_creation(): |
|
|
"""Test graph edge creation""" |
|
|
edge = GraphEdge( |
|
|
from_node="node1", |
|
|
to_node="node2", |
|
|
relation=RelationType.USES, |
|
|
confidence=0.8, |
|
|
supporting_chunks=[] |
|
|
) |
|
|
|
|
|
assert edge.from_node == "node1" |
|
|
assert edge.to_node == "node2" |
|
|
assert edge.relation == RelationType.USES |
|
|
assert edge.confidence == 0.8 |
|
|
|
|
|
|
|
|
def test_triple_creation(): |
|
|
"""Test triple model""" |
|
|
triple = Triple( |
|
|
subject="Machine Learning", |
|
|
predicate="uses", |
|
|
object="Neural Networks", |
|
|
confidence=0.9, |
|
|
page_number=5 |
|
|
) |
|
|
|
|
|
assert triple.subject == "Machine Learning" |
|
|
assert triple.predicate == "uses" |
|
|
assert triple.object == "Neural Networks" |
|
|
assert triple.confidence == 0.9 |
|
|
|
|
|
|
|
|
def test_settings_load(): |
|
|
"""Test configuration loading""" |
|
|
assert settings.app_name == "GraphLLM" |
|
|
assert settings.chunk_size > 0 |
|
|
assert settings.embedding_model is not None |
|
|
|
|
|
|
|
|
@pytest.mark.asyncio |
|
|
async def test_pdf_processor_import(): |
|
|
"""Test PDF processor can be imported""" |
|
|
from pdf_processor import PDFProcessor |
|
|
processor = PDFProcessor() |
|
|
assert processor is not None |
|
|
|
|
|
|
|
|
@pytest.mark.asyncio |
|
|
async def test_embedding_service_import(): |
|
|
"""Test embedding service can be imported""" |
|
|
from embedding_service import EmbeddingService |
|
|
|
|
|
|
|
|
|
|
|
pass |
|
|
|
|
|
|
|
|
@pytest.mark.asyncio |
|
|
async def test_graph_store_import(): |
|
|
"""Test graph store can be imported""" |
|
|
from graph_store import GraphStore |
|
|
store = GraphStore(use_neo4j=False) |
|
|
assert store is not None |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
pytest.main([__file__, "-v"]) |
|
|
|
