From 2495efe415c05a22ddac675eea266dffa965e144 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Fri, 5 Sep 2025 15:58:40 +0100 Subject: [PATCH] Update tests --- .../test_object_extraction_integration.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/integration/test_object_extraction_integration.py b/tests/integration/test_object_extraction_integration.py index b54b559a..7b2245ce 100644 --- a/tests/integration/test_object_extraction_integration.py +++ b/tests/integration/test_object_extraction_integration.py @@ -270,9 +270,9 @@ class TestObjectExtractionServiceIntegration: assert len(customer_calls) == 1 customer_obj = customer_calls[0] - assert customer_obj.values["customer_id"] == "CUST001" - assert customer_obj.values["name"] == "John Smith" - assert customer_obj.values["email"] == "john.smith@email.com" + assert customer_obj.values[0]["customer_id"] == "CUST001" + assert customer_obj.values[0]["name"] == "John Smith" + assert customer_obj.values[0]["email"] == "john.smith@email.com" assert customer_obj.confidence > 0.5 @pytest.mark.asyncio @@ -335,10 +335,10 @@ class TestObjectExtractionServiceIntegration: assert len(product_calls) == 1 product_obj = product_calls[0] - assert product_obj.values["product_id"] == "PROD001" - assert product_obj.values["name"] == "Gaming Laptop" - assert product_obj.values["price"] == "1299.99" - assert product_obj.values["category"] == "electronics" + assert product_obj.values[0]["product_id"] == "PROD001" + assert product_obj.values[0]["name"] == "Gaming Laptop" + assert product_obj.values[0]["price"] == "1299.99" + assert product_obj.values[0]["category"] == "electronics" @pytest.mark.asyncio async def test_concurrent_extraction_integration(self, integration_config, mock_integrated_flow):