Added a helper to assert on lines
Adapted test to use new helper

Still a bit clunky to write, but its getting better.