Adding test for set comprehension.