Add model definition tests