Skip to content

Commit

Permalink
Fix quotes
Browse files Browse the repository at this point in the history
  • Loading branch information
lajohn4747 committed Mar 6, 2024
1 parent 33ec937 commit 852e684
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/unit/test_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,7 @@ def test_benchmark_single_table_with_timeout(mock_multiprocessing, mock__score):
})
pd.testing.assert_frame_equal(scores, expected_scores)


@patch('sdgym.benchmark.boto3.session.Session')
@patch('sdgym.benchmark._create_instance_on_ec2')
def test_run_ec2_flag(create_ec2_mock, session_mock):
Expand Down Expand Up @@ -115,6 +116,7 @@ def test_run_ec2_flag(create_ec2_mock, session_mock):
# Assert
create_ec2_mock.assert_called_once()


@patch('sdgym.benchmark.boto3.session.Session')
def test__create_sdgym_script(session_mock):
session_mock.get_credentials.return_value = MagicMock()
Expand Down Expand Up @@ -145,6 +147,6 @@ def test__create_sdgym_script(session_mock):
assert 'detailed_results_folder=None' in result
assert 'multi_processing_config=None' in result
assert "sdmetrics=[('NewRowSynthesis', {'synthetic_sample_size': 1000})]" in result
assert "timeout=600" in result
assert 'timeout=600' in result
assert 'compute_quality_score=False' in result
assert 'import boto3' in result

0 comments on commit 852e684

Please sign in to comment.