mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-09 15:24:46 +08:00
3204a39f6c
* end_pos calculation didn't respect multiline tokens * replaced all index access to Token * wrapped all code that injects token tuples with Token.from_tuple() * repr of Token is still its tuple form!? * PEP8 where I read or wrote code
26 lines
640 B
Python
26 lines
640 B
Python
import jedi.parser as parser
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError: # pragma: no cover
|
|
import unittest
|
|
|
|
|
|
class TokenTest(unittest.TestCase):
|
|
def test_end_pos_one_line(self):
|
|
parsed = parser.Parser('''
|
|
def testit():
|
|
a = "huhu"
|
|
''')
|
|
tok = parsed.top_module.subscopes[0].statements[0].token_list[2]
|
|
self.assertEqual(tok.end_pos, (3, 14))
|
|
|
|
def test_end_pos_multi_line(self):
|
|
parsed = parser.Parser('''
|
|
def testit():
|
|
a = """huhu
|
|
asdfasdf""" + "h"
|
|
''')
|
|
tok = parsed.top_module.subscopes[0].statements[0].token_list[2]
|
|
self.assertEqual(tok.end_pos, (4, 11))
|