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

Fixing tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-03-09 10:05:39 +00:00
parent dcee1b8de2
commit 4e3db11323
2 changed files with 64 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/contract/test_structured_data_contracts.py
View file

@ -285,11 +285,11 @@ class TestStructuredEmbeddingsContracts:
collection="test_collection",
metadata=[]
)
# Act
embedding = StructuredObjectEmbedding(
metadata=metadata,
vectors=[[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]],
vector=[0.1, 0.2, 0.3],
schema_name="customer_records",
object_id="customer_123",
field_embeddings={
@ -301,7 +301,7 @@ class TestStructuredEmbeddingsContracts:
# Assert
assert embedding.schema_name == "customer_records"
assert embedding.object_id == "customer_123"
assert len(embedding.vectors) == 2
assert len(embedding.vector) == 3
assert len(embedding.field_embeddings) == 2
assert "name" in embedding.field_embeddings