added tests for dictionary comprehension and test comprehension