def test_release_raises_error(self, mocker):
"""Check the error when `release` raises an exception"""
mocked_release = mocker.Mock()
error_message = "File not found"
mocked_release.side_effect = FileNotFoundError(error_message)
data_set = LambdaDataSet(None, None, None, mocked_release)
with pytest.raises(DataSetError, match=error_message):
data_set.release()
mocked_release.assert_called_once_with()
def test_release_not_implemented(self):
"""Check that `release` does nothing by default"""
data_set = LambdaDataSet(None, None)
data_set.release()
def test_release_invocation(self, mocker):
"""Test the basic `release` method invocation"""
mocked_release = mocker.Mock()
data_set = LambdaDataSet(None, None, None, mocked_release)
data_set.release()
mocked_release.assert_called_once_with()
