def test_load_raises_error(self): """Check the error if loading the LambdaDataSet raises an exception""" error_message = "Internal load exception message" def internal_load(): raise FileNotFoundError(error_message) data_set = LambdaDataSet(internal_load, None) with pytest.raises(DataSetError, match=error_message): data_set.load()
def test_load_invocation(self, mocker): """Test the basic `load` method invocation""" mocked_load = mocker.Mock(return_value=42) data_set = LambdaDataSet(mocked_load, None) result = data_set.load() mocked_load.assert_called_once_with() assert result == 42