Add test_functional_models.py