Don't place all of the list comprehension on separate lines.

If one part of a list comp goes over 79 columns, it could place the list
comp on separate lines.
Temporarily fix testcase.
There may be stuff after the container to take into account for the size.