Add tests for comprehension testing.