Make the assertion about the AST coming from a parse consider all lines,
not just the first, and fix tests accordingly.