Refactor the tests so that they can be applied to more models.