Iran's ambassador to Australia has been seen preparing to leave the country as a result of a diplomatic rift between the two countries.