Skip to content

Commit b03bd09

Browse files
committed
add return value
1 parent cbafe41 commit b03bd09

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/unit/test_benchmark.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)