Parameterized the iteration limit in the unit test.