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.
This commit is contained in:
@@ -93,12 +93,12 @@ async def test_run_none_path_flag(workflow_mock, format_and_export_prediction):
|
|||||||
{
|
{
|
||||||
'schema': input_data['schema'],
|
'schema': input_data['schema'],
|
||||||
'table_name': input_data['table_name'],
|
'table_name': input_data['table_name'],
|
||||||
'data': workflow_mock.execute_local_activity_method.return_value,
|
'data': prediction_data,
|
||||||
**metadata,
|
|
||||||
'timestamp_conversion': {
|
'timestamp_conversion': {
|
||||||
'column': 'timestamp',
|
'column': 'timestamp',
|
||||||
'format': DATETIME_FORMAT_WITH_TZ,
|
'format': DATETIME_FORMAT_WITH_TZ,
|
||||||
},
|
},
|
||||||
|
**metadata,
|
||||||
},
|
},
|
||||||
retry_policy=ANY,
|
retry_policy=ANY,
|
||||||
start_to_close_timeout=ANY,
|
start_to_close_timeout=ANY,
|
||||||
|
|||||||
Reference in New Issue
Block a user