File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -1506,9 +1506,11 @@ def test_benchmark_multi_table_aws_no_jobs(
15061506 mock_validate_output_destination .return_value = 's3_client_mock'
15071507 mock_generate_job_args_list .return_value = []
15081508 mock_import_and_validate_synthesizers .return_value = synthesizers
1509+ expected_result = pd .DataFrame ({'Dataset' : []})
1510+ mock_get_empty_dataframe .return_value = expected_result
15091511
15101512 # Run
1511- benchmark_multi_table_aws (
1513+ result = benchmark_multi_table_aws (
15121514 output_destination = output_destination ,
15131515 aws_access_key_id = aws_access_key_id ,
15141516 aws_secret_access_key = aws_secret_access_key ,
@@ -1517,6 +1519,7 @@ def test_benchmark_multi_table_aws_no_jobs(
15171519 )
15181520
15191521 # Assert
1522+ pd .testing .assert_frame_equal (result , expected_result )
15201523 assert 'MultiTableUniformSynthesizer' in synthesizers
15211524 mock_validate_output_destination .assert_called_once_with (
15221525 output_destination ,
You can’t perform that action at this time.
0 commit comments