Added more tests for the Iteration model.

Signed-off-by: Tushar Narayan <tusharnarayan@gmail.com>