Fix integration tests for edx solutions

@e-kolpakov Good point. I will change it. I actually prefer to use the one that is more readable, but python comprehensions are generally nice.