add test_model.py