def test_multiple_line(self): line = 'This is a multi line\nSee, two lines' self.assertEqual(20, end_of_line(line, 2))
def test_single_line(self): line = 'This is a single line.' self.assertEqual(len(line), end_of_line(line, 2))