def test_call_with_dataset_name(self, mocker, mocks_init_class):
mock_run = mocker.patch.object(DownloadAPI, "run")
download("some_dataset")
assert_mock_init_class(mocks_init_class)
assert mock_run.called
def test_call_with_all_input_args(self, mocker, mocks_init_class,
test_data):
mock_run = mocker.patch.object(DownloadAPI, "run")
download(test_data["dataset"], test_data["data_dir"],
test_data["extract_data"], test_data["verbose"])
assert_mock_init_class(mocks_init_class)
assert mock_run.called
def test_call__raises_error_too_many_args(self, mocker):
with pytest.raises(TypeError):
download('some_dataset', os.path.join('some', 'dir', 'path'), True,
True, 'extra_field')
def test_call__raise_error_invalid_dataset_name(self, mocker,
mocks_init_class):
mock_run = mocker.patch.object(DownloadAPI, "run")
with pytest.raises(AssertionError):
download(None)
def test_call__raises_error_no_input_args(self, mocker, mocks_init_class):
mock_run = mocker.patch.object(DownloadAPI, "run")
with pytest.raises(TypeError):
download()