Test generate_models
Adding a Dataset-like API