From a5a869aa7d89860d903cfe4df170d7faa0a54ab6 Mon Sep 17 00:00:00 2001 From: Samuel Fendell Date: Tue, 26 Mar 2024 16:26:39 -0700 Subject: [PATCH] [PLT-383] Fix SDK integration tests. --- tests/integration/annotation_import/test_data_types.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/integration/annotation_import/test_data_types.py b/tests/integration/annotation_import/test_data_types.py index eaf71e55d..924832a86 100644 --- a/tests/integration/annotation_import/test_data_types.py +++ b/tests/integration/annotation_import/test_data_types.py @@ -296,7 +296,9 @@ def test_import_data_types_v2(client, configured_project, initial_dataset, # to be similar to tests/integration/test_task_queue.py result = export_v2_test_helpers.run_project_export_v2_task(project) - exported_data = result[0] + find_data_row = lambda dr: dr['data_row']['id'] == data_row.uid + exported_data = list(filter(find_data_row, result))[0] + assert exported_data # timestamp fields are in iso format validate_iso_format(exported_data['data_row']['details']['created_at'])