add a list comprehension to generators tests