tests: add test_comprehension.nit