예제 #1
0
def test_api_model_names():
    """
    Verify the expected API model list.

    This allows parametrized testing of the API Models, for better failure messages.
    """
    names = [name for model, name in api_models_and_names()]
    assert names == API_MODEL_NAMES
예제 #2
0
 def setup_class(cls):
     """Generate the dictionary of model names to model classes."""
     for model, model_name in api_models_and_names():
         cls.MODEL[model_name] = model