week 4 assign: tests for the iterator