mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
parent
a8e437fc7f
commit
6c7af8789d
216 changed files with 31360 additions and 1611 deletions
|
|
@ -317,12 +317,12 @@ class TestObjectExtractionBusinessLogic:
|
|||
metadata=[]
|
||||
)
|
||||
|
||||
values = {
|
||||
values = [{
|
||||
"customer_id": "CUST001",
|
||||
"name": "John Doe",
|
||||
"email": "john@example.com",
|
||||
"status": "active"
|
||||
}
|
||||
}]
|
||||
|
||||
# Act
|
||||
extracted_obj = ExtractedObject(
|
||||
|
|
@ -335,7 +335,7 @@ class TestObjectExtractionBusinessLogic:
|
|||
|
||||
# Assert
|
||||
assert extracted_obj.schema_name == "customer_records"
|
||||
assert extracted_obj.values["customer_id"] == "CUST001"
|
||||
assert extracted_obj.values[0]["customer_id"] == "CUST001"
|
||||
assert extracted_obj.confidence == 0.95
|
||||
assert "John Doe" in extracted_obj.source_span
|
||||
assert extracted_obj.metadata.user == "test_user"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue