Add test for __iter__