def test_seek_first_line_of_text():
text = io.StringIO("one\ntwo\nthree")
assert_equal("one", textseek.seek_line(text, 1))
def _print_location(location):
print("File '{}', line {}, col {}".format(location.filename, location.lineno, location.offset))
print()
with open(location.filename) as source_file:
print(" " + textseek.seek_line(source_file, location.lineno))
print(" " * (2 + location.offset) + "^")
def test_seek_line_of_text_in_middle():
text = io.StringIO("one\ntwo\nthree")
assert_equal("two", textseek.seek_line(text, 2))
