apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-22 03:31:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Added embeddings to tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-07-14 15:57:39 +01:00
parent f7437cf72d
commit c765859f00
6 changed files with 15 additions and 1230 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/unit/test_embeddings/test_embedding_utils.py
View file

@ -83,7 +83,7 @@ class TestEmbeddingDimensionConsistency:
for i, result in enumerate(results):
assert len(result) == dimension, f"Text length {len(test_texts[i])} produced wrong dimension"
async def test_dimension_validation_different_models(self):
def test_dimension_validation_different_models(self):
"""Test dimension validation for different model configurations"""
# Arrange
models_and_dims = [
@ -92,20 +92,11 @@ class TestEmbeddingDimensionConsistency:
("large-model", 1536)
]
# Act & Assert
for model_name, expected_dim in models_and_dims:
def mock_embedding(text):
return [0.1] * expected_dim
processor = MockEmbeddingProcessor(
model=model_name,
embedding_function=mock_embedding
)
# Act
result = await processor.on_embeddings("Test text")
# Assert
assert len(result) == expected_dim, f"Model {model_name} produced wrong dimension"
# Test dimension validation logic
test_vector = [0.1] * expected_dim
assert len(test_vector) == expected_dim, f"Model {model_name} dimension mismatch"
class TestEmbeddingBatchProcessing: