Should countries respond to diplomatic expulsions with matching expulsions?

Iran expelled a Swedish diplomat after Sweden sent home an Iranian embassy official.