Ejemplo n.º 1
0
    def test_get_lines(self):
        SRC = SourceString('hello\nworld\nthis\nis\na\ntest')

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

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

        assert SRC.get_lines(10, 20) == None