From c1a5ab3d3b7f4583b65dadd24a1dd5cf5fac240d Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 23 Oct 2025 17:55:35 -0300 Subject: [PATCH] SIENTIAPDE-1314 Update test_format_and_export_prediction.py to use prediction_data for workflow execution - Replaced the mock return value with prediction_data in the test_run_none_path_flag function to ensure accurate testing of the workflow's data handling. --- .../subworkflows/test_format_and_export_prediction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py b/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py index 59e39b3..ecd3bc2 100644 --- a/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py +++ b/tests/laborious/workflows/subworkflows/test_format_and_export_prediction.py @@ -93,12 +93,12 @@ async def test_run_none_path_flag(workflow_mock, format_and_export_prediction): { 'schema': input_data['schema'], 'table_name': input_data['table_name'], - 'data': workflow_mock.execute_local_activity_method.return_value, - **metadata, + 'data': prediction_data, 'timestamp_conversion': { 'column': 'timestamp', 'format': DATETIME_FORMAT_WITH_TZ, }, + **metadata, }, retry_policy=ANY, start_to_close_timeout=ANY,