add tests for models.py