Ejemplos de SourceString.get_surrounding_lines en Python

Ejemplo n.º 1

Archivo: test_sourcestring.py Proyecto: Nekroze/partpy

    def test_get_surrounding_lines(self):
        SRC = SourceString('hello\nworld\nthis\nis\na\ntest')

        lines = [str(x) for x in SRC.get_surrounding_lines()]
        assert lines == ['1   |hello\n', '2   |world\n']
        lines = ''.join([repr(x) for x in SRC.get_surrounding_lines()])
        assert lines == 'hello\nworld\n'

        SRC.eat_string('hello\nworld\n')

        lines = [str(x) for x in SRC.get_surrounding_lines()]
        assert lines == ['2   |world\n', '3   |this\n', '4   |is\n']
        lines = ''.join([repr(x) for x in SRC.get_surrounding_lines()])
        assert lines == 'world\nthis\nis\n'

        lines = [str(x) for x in SRC.get_surrounding_lines(1, 0)]
        assert lines == ['2   |world\n', '3   |this\n']
        lines = ''.join([repr(x) for x in SRC.get_surrounding_lines(1, 0)])
        assert lines == 'world\nthis\n'

        SRC.eat_string('this\nis\na\n')
        lines = [str(x) for x in SRC.get_surrounding_lines()]
        assert lines == ['5   |a\n', '6   |test']
        lines = ''.join([repr(x) for x in SRC.get_surrounding_lines()])
        assert lines == 'a\ntest'