Move relation tests models into models.py