

def test_split_lines_2(self):
    lines = list(split_lines([(Token.A, 'line1'), (Token.B, 'line2\nline3\nline4')]))
    self.assertEqual(lines, [[(Token.A, 'line1'), (Token.B, 'line2')], [(Token.B, 'line3')], [(Token.B, 'line4')]])

dataset/ETHPy150Open jonathanslenders/python-prompt-toolkit/tests/layout_tests/__init__.py SplitLinesTest.test_split_lines_2/original