Add tests for the soft masses after the iteration