Set Comprehension Extension for z3 in Python