Simplify test_models.py