Add tests for editIteration function