updated tests for lambda update (some comprehensions stuff might be changed)