Add test cases

* Enum/Stream
* Comprehension