470 lines
18 KiB
Python
470 lines
18 KiB
Python
"""Unittest for idlelib.pyparse.py.
|
|
|
|
Coverage: 97%
|
|
"""
|
|
|
|
from collections import namedtuple
|
|
import unittest
|
|
from idlelib import pyparse
|
|
|
|
|
|
class ParseMapTest(unittest.TestCase):
|
|
|
|
def test_parsemap(self):
|
|
keepwhite = {ord(c): ord(c) for c in ' \t\n\r'}
|
|
mapping = pyparse.ParseMap(keepwhite)
|
|
self.assertEqual(mapping[ord('\t')], ord('\t'))
|
|
self.assertEqual(mapping[ord('a')], ord('x'))
|
|
self.assertEqual(mapping[1000], ord('x'))
|
|
|
|
def test_trans(self):
|
|
# trans is the production instance of ParseMap, used in _study1
|
|
parser = pyparse.Parser(4, 4)
|
|
self.assertEqual('\t a([{b}])b"c\'d\n'.translate(pyparse.trans),
|
|
'xxx(((x)))x"x\'x\n')
|
|
|
|
|
|
class PyParseTest(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.parser = pyparse.Parser(indentwidth=4, tabwidth=4)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
del cls.parser
|
|
|
|
def test_init(self):
|
|
self.assertEqual(self.parser.indentwidth, 4)
|
|
self.assertEqual(self.parser.tabwidth, 4)
|
|
|
|
def test_set_code(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
|
|
# Not empty and doesn't end with newline.
|
|
with self.assertRaises(AssertionError):
|
|
setcode('a')
|
|
|
|
tests = ('',
|
|
'a\n')
|
|
|
|
for string in tests:
|
|
with self.subTest(string=string):
|
|
setcode(string)
|
|
eq(p.code, string)
|
|
eq(p.study_level, 0)
|
|
|
|
def test_find_good_parse_start(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
start = p.find_good_parse_start
|
|
|
|
# Split def across lines.
|
|
setcode('"""This is a module docstring"""\n'
|
|
'class C():\n'
|
|
' def __init__(self, a,\n'
|
|
' b=True):\n'
|
|
' pass\n'
|
|
)
|
|
|
|
# No value sent for is_char_in_string().
|
|
self.assertIsNone(start())
|
|
|
|
# Make text look like a string. This returns pos as the start
|
|
# position, but it's set to None.
|
|
self.assertIsNone(start(is_char_in_string=lambda index: True))
|
|
|
|
# Make all text look like it's not in a string. This means that it
|
|
# found a good start position.
|
|
eq(start(is_char_in_string=lambda index: False), 44)
|
|
|
|
# If the beginning of the def line is not in a string, then it
|
|
# returns that as the index.
|
|
eq(start(is_char_in_string=lambda index: index > 44), 44)
|
|
# If the beginning of the def line is in a string, then it
|
|
# looks for a previous index.
|
|
eq(start(is_char_in_string=lambda index: index >= 44), 33)
|
|
# If everything before the 'def' is in a string, then returns None.
|
|
# The non-continuation def line returns 44 (see below).
|
|
eq(start(is_char_in_string=lambda index: index < 44), None)
|
|
|
|
# Code without extra line break in def line - mostly returns the same
|
|
# values.
|
|
setcode('"""This is a module docstring"""\n'
|
|
'class C():\n'
|
|
' def __init__(self, a, b=True):\n'
|
|
' pass\n'
|
|
)
|
|
eq(start(is_char_in_string=lambda index: False), 44)
|
|
eq(start(is_char_in_string=lambda index: index > 44), 44)
|
|
eq(start(is_char_in_string=lambda index: index >= 44), 33)
|
|
# When the def line isn't split, this returns which doesn't match the
|
|
# split line test.
|
|
eq(start(is_char_in_string=lambda index: index < 44), 44)
|
|
|
|
def test_set_lo(self):
|
|
code = (
|
|
'"""This is a module docstring"""\n'
|
|
'class C():\n'
|
|
' def __init__(self, a,\n'
|
|
' b=True):\n'
|
|
' pass\n'
|
|
)
|
|
p = self.parser
|
|
p.set_code(code)
|
|
|
|
# Previous character is not a newline.
|
|
with self.assertRaises(AssertionError):
|
|
p.set_lo(5)
|
|
|
|
# A value of 0 doesn't change self.code.
|
|
p.set_lo(0)
|
|
self.assertEqual(p.code, code)
|
|
|
|
# An index that is preceded by a newline.
|
|
p.set_lo(44)
|
|
self.assertEqual(p.code, code[44:])
|
|
|
|
def test_study1(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
study = p._study1
|
|
|
|
(NONE, BACKSLASH, FIRST, NEXT, BRACKET) = range(5)
|
|
TestInfo = namedtuple('TestInfo', ['string', 'goodlines',
|
|
'continuation'])
|
|
tests = (
|
|
TestInfo('', [0], NONE),
|
|
# Docstrings.
|
|
TestInfo('"""This is a complete docstring."""\n', [0, 1], NONE),
|
|
TestInfo("'''This is a complete docstring.'''\n", [0, 1], NONE),
|
|
TestInfo('"""This is a continued docstring.\n', [0, 1], FIRST),
|
|
TestInfo("'''This is a continued docstring.\n", [0, 1], FIRST),
|
|
TestInfo('"""Closing quote does not match."\n', [0, 1], FIRST),
|
|
TestInfo('"""Bracket in docstring [\n', [0, 1], FIRST),
|
|
TestInfo("'''Incomplete two line docstring.\n\n", [0, 2], NEXT),
|
|
# Single-quoted strings.
|
|
TestInfo('"This is a complete string."\n', [0, 1], NONE),
|
|
TestInfo('"This is an incomplete string.\n', [0, 1], NONE),
|
|
TestInfo("'This is more incomplete.\n\n", [0, 1, 2], NONE),
|
|
# Comment (backslash does not continue comments).
|
|
TestInfo('# Comment\\\n', [0, 1], NONE),
|
|
# Brackets.
|
|
TestInfo('("""Complete string in bracket"""\n', [0, 1], BRACKET),
|
|
TestInfo('("""Open string in bracket\n', [0, 1], FIRST),
|
|
TestInfo('a = (1 + 2) - 5 *\\\n', [0, 1], BACKSLASH), # No bracket.
|
|
TestInfo('\n def function1(self, a,\n b):\n',
|
|
[0, 1, 3], NONE),
|
|
TestInfo('\n def function1(self, a,\\\n', [0, 1, 2], BRACKET),
|
|
TestInfo('\n def function1(self, a,\n', [0, 1, 2], BRACKET),
|
|
TestInfo('())\n', [0, 1], NONE), # Extra closer.
|
|
TestInfo(')(\n', [0, 1], BRACKET), # Extra closer.
|
|
# For the mismatched example, it doesn't look like contination.
|
|
TestInfo('{)(]\n', [0, 1], NONE), # Mismatched.
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string) # resets study_level
|
|
study()
|
|
eq(p.study_level, 1)
|
|
eq(p.goodlines, test.goodlines)
|
|
eq(p.continuation, test.continuation)
|
|
|
|
# Called again, just returns without reprocessing.
|
|
self.assertIsNone(study())
|
|
|
|
def test_get_continuation_type(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
gettype = p.get_continuation_type
|
|
|
|
(NONE, BACKSLASH, FIRST, NEXT, BRACKET) = range(5)
|
|
TestInfo = namedtuple('TestInfo', ['string', 'continuation'])
|
|
tests = (
|
|
TestInfo('', NONE),
|
|
TestInfo('"""This is a continuation docstring.\n', FIRST),
|
|
TestInfo("'''This is a multiline-continued docstring.\n\n", NEXT),
|
|
TestInfo('a = (1 + 2) - 5 *\\\n', BACKSLASH),
|
|
TestInfo('\n def function1(self, a,\\\n', BRACKET)
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
eq(gettype(), test.continuation)
|
|
|
|
def test_study2(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
study = p._study2
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'start', 'end', 'lastch',
|
|
'openbracket', 'bracketing'])
|
|
tests = (
|
|
TestInfo('', 0, 0, '', None, ((0, 0),)),
|
|
TestInfo("'''This is a multiline continutation docstring.\n\n",
|
|
0, 49, "'", None, ((0, 0), (0, 1), (49, 0))),
|
|
TestInfo(' # Comment\\\n',
|
|
0, 12, '', None, ((0, 0), (1, 1), (12, 0))),
|
|
# A comment without a space is a special case
|
|
TestInfo(' #Comment\\\n',
|
|
0, 0, '', None, ((0, 0),)),
|
|
# Backslash continuation.
|
|
TestInfo('a = (1 + 2) - 5 *\\\n',
|
|
0, 19, '*', None, ((0, 0), (4, 1), (11, 0))),
|
|
# Bracket continuation with close.
|
|
TestInfo('\n def function1(self, a,\n b):\n',
|
|
1, 48, ':', None, ((1, 0), (17, 1), (46, 0))),
|
|
# Bracket continuation with unneeded backslash.
|
|
TestInfo('\n def function1(self, a,\\\n',
|
|
1, 28, ',', 17, ((1, 0), (17, 1))),
|
|
# Bracket continuation.
|
|
TestInfo('\n def function1(self, a,\n',
|
|
1, 27, ',', 17, ((1, 0), (17, 1))),
|
|
# Bracket continuation with comment at end of line with text.
|
|
TestInfo('\n def function1(self, a, # End of line comment.\n',
|
|
1, 51, ',', 17, ((1, 0), (17, 1), (28, 2), (51, 1))),
|
|
# Multi-line statement with comment line in between code lines.
|
|
TestInfo(' a = ["first item",\n # Comment line\n "next item",\n',
|
|
0, 55, ',', 6, ((0, 0), (6, 1), (7, 2), (19, 1),
|
|
(23, 2), (38, 1), (42, 2), (53, 1))),
|
|
TestInfo('())\n',
|
|
0, 4, ')', None, ((0, 0), (0, 1), (2, 0), (3, 0))),
|
|
TestInfo(')(\n', 0, 3, '(', 1, ((0, 0), (1, 0), (1, 1))),
|
|
# Wrong closers still decrement stack level.
|
|
TestInfo('{)(]\n',
|
|
0, 5, ']', None, ((0, 0), (0, 1), (2, 0), (2, 1), (4, 0))),
|
|
# Character after backslash.
|
|
TestInfo(':\\a\n', 0, 4, '\\a', None, ((0, 0),)),
|
|
TestInfo('\n', 0, 0, '', None, ((0, 0),)),
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
study()
|
|
eq(p.study_level, 2)
|
|
eq(p.stmt_start, test.start)
|
|
eq(p.stmt_end, test.end)
|
|
eq(p.lastch, test.lastch)
|
|
eq(p.lastopenbracketpos, test.openbracket)
|
|
eq(p.stmt_bracketing, test.bracketing)
|
|
|
|
# Called again, just returns without reprocessing.
|
|
self.assertIsNone(study())
|
|
|
|
def test_get_num_lines_in_stmt(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
getlines = p.get_num_lines_in_stmt
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'lines'])
|
|
tests = (
|
|
TestInfo('[x for x in a]\n', 1), # Closed on one line.
|
|
TestInfo('[x\nfor x in a\n', 2), # Not closed.
|
|
TestInfo('[x\\\nfor x in a\\\n', 2), # "", uneeded backslashes.
|
|
TestInfo('[x\nfor x in a\n]\n', 3), # Closed on multi-line.
|
|
TestInfo('\n"""Docstring comment L1"""\nL2\nL3\nL4\n', 1),
|
|
TestInfo('\n"""Docstring comment L1\nL2"""\nL3\nL4\n', 1),
|
|
TestInfo('\n"""Docstring comment L1\\\nL2\\\nL3\\\nL4\\\n', 4),
|
|
TestInfo('\n\n"""Docstring comment L1\\\nL2\\\nL3\\\nL4\\\n"""\n', 5)
|
|
)
|
|
|
|
# Blank string doesn't have enough elements in goodlines.
|
|
setcode('')
|
|
with self.assertRaises(IndexError):
|
|
getlines()
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
eq(getlines(), test.lines)
|
|
|
|
def test_compute_bracket_indent(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
indent = p.compute_bracket_indent
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'spaces'])
|
|
tests = (
|
|
TestInfo('def function1(self, a,\n', 14),
|
|
# Characters after bracket.
|
|
TestInfo('\n def function1(self, a,\n', 18),
|
|
TestInfo('\n\tdef function1(self, a,\n', 18),
|
|
# No characters after bracket.
|
|
TestInfo('\n def function1(\n', 8),
|
|
TestInfo('\n\tdef function1(\n', 8),
|
|
TestInfo('\n def function1( \n', 8), # Ignore extra spaces.
|
|
TestInfo('[\n"first item",\n # Comment line\n "next item",\n', 0),
|
|
TestInfo('[\n "first item",\n # Comment line\n "next item",\n', 2),
|
|
TestInfo('["first item",\n # Comment line\n "next item",\n', 1),
|
|
TestInfo('(\n', 4),
|
|
TestInfo('(a\n', 1),
|
|
)
|
|
|
|
# Must be C_BRACKET continuation type.
|
|
setcode('def function1(self, a, b):\n')
|
|
with self.assertRaises(AssertionError):
|
|
indent()
|
|
|
|
for test in tests:
|
|
setcode(test.string)
|
|
eq(indent(), test.spaces)
|
|
|
|
def test_compute_backslash_indent(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
indent = p.compute_backslash_indent
|
|
|
|
# Must be C_BACKSLASH continuation type.
|
|
errors = (('def function1(self, a, b\\\n'), # Bracket.
|
|
(' """ (\\\n'), # Docstring.
|
|
('a = #\\\n'), # Inline comment.
|
|
)
|
|
for string in errors:
|
|
with self.subTest(string=string):
|
|
setcode(string)
|
|
with self.assertRaises(AssertionError):
|
|
indent()
|
|
|
|
TestInfo = namedtuple('TestInfo', ('string', 'spaces'))
|
|
tests = (TestInfo('a = (1 + 2) - 5 *\\\n', 4),
|
|
TestInfo('a = 1 + 2 - 5 *\\\n', 4),
|
|
TestInfo(' a = 1 + 2 - 5 *\\\n', 8),
|
|
TestInfo(' a = "spam"\\\n', 6),
|
|
TestInfo(' a = \\\n"a"\\\n', 4),
|
|
TestInfo(' a = #\\\n"a"\\\n', 5),
|
|
TestInfo('a == \\\n', 2),
|
|
TestInfo('a != \\\n', 2),
|
|
# Difference between containing = and those not.
|
|
TestInfo('\\\n', 2),
|
|
TestInfo(' \\\n', 6),
|
|
TestInfo('\t\\\n', 6),
|
|
TestInfo('a\\\n', 3),
|
|
TestInfo('{}\\\n', 4),
|
|
TestInfo('(1 + 2) - 5 *\\\n', 3),
|
|
)
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
eq(indent(), test.spaces)
|
|
|
|
def test_get_base_indent_string(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
baseindent = p.get_base_indent_string
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'indent'])
|
|
tests = (TestInfo('', ''),
|
|
TestInfo('def a():\n', ''),
|
|
TestInfo('\tdef a():\n', '\t'),
|
|
TestInfo(' def a():\n', ' '),
|
|
TestInfo(' def a(\n', ' '),
|
|
TestInfo('\t\n def a(\n', ' '),
|
|
TestInfo('\t\n # Comment.\n', ' '),
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
eq(baseindent(), test.indent)
|
|
|
|
def test_is_block_opener(self):
|
|
yes = self.assertTrue
|
|
no = self.assertFalse
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
opener = p.is_block_opener
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'assert_'])
|
|
tests = (
|
|
TestInfo('def a():\n', yes),
|
|
TestInfo('\n def function1(self, a,\n b):\n', yes),
|
|
TestInfo(':\n', yes),
|
|
TestInfo('a:\n', yes),
|
|
TestInfo('):\n', yes),
|
|
TestInfo('(:\n', yes),
|
|
TestInfo('":\n', no),
|
|
TestInfo('\n def function1(self, a,\n', no),
|
|
TestInfo('def function1(self, a):\n pass\n', no),
|
|
TestInfo('# A comment:\n', no),
|
|
TestInfo('"""A docstring:\n', no),
|
|
TestInfo('"""A docstring:\n', no),
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
test.assert_(opener())
|
|
|
|
def test_is_block_closer(self):
|
|
yes = self.assertTrue
|
|
no = self.assertFalse
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
closer = p.is_block_closer
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'assert_'])
|
|
tests = (
|
|
TestInfo('return\n', yes),
|
|
TestInfo('\tbreak\n', yes),
|
|
TestInfo(' continue\n', yes),
|
|
TestInfo(' raise\n', yes),
|
|
TestInfo('pass \n', yes),
|
|
TestInfo('pass\t\n', yes),
|
|
TestInfo('return #\n', yes),
|
|
TestInfo('raised\n', no),
|
|
TestInfo('returning\n', no),
|
|
TestInfo('# return\n', no),
|
|
TestInfo('"""break\n', no),
|
|
TestInfo('"continue\n', no),
|
|
TestInfo('def function1(self, a):\n pass\n', yes),
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
test.assert_(closer())
|
|
|
|
def test_get_last_stmt_bracketing(self):
|
|
eq = self.assertEqual
|
|
p = self.parser
|
|
setcode = p.set_code
|
|
bracketing = p.get_last_stmt_bracketing
|
|
|
|
TestInfo = namedtuple('TestInfo', ['string', 'bracket'])
|
|
tests = (
|
|
TestInfo('', ((0, 0),)),
|
|
TestInfo('a\n', ((0, 0),)),
|
|
TestInfo('()()\n', ((0, 0), (0, 1), (2, 0), (2, 1), (4, 0))),
|
|
TestInfo('(\n)()\n', ((0, 0), (0, 1), (3, 0), (3, 1), (5, 0))),
|
|
TestInfo('()\n()\n', ((3, 0), (3, 1), (5, 0))),
|
|
TestInfo('()(\n)\n', ((0, 0), (0, 1), (2, 0), (2, 1), (5, 0))),
|
|
TestInfo('(())\n', ((0, 0), (0, 1), (1, 2), (3, 1), (4, 0))),
|
|
TestInfo('(\n())\n', ((0, 0), (0, 1), (2, 2), (4, 1), (5, 0))),
|
|
# Same as matched test.
|
|
TestInfo('{)(]\n', ((0, 0), (0, 1), (2, 0), (2, 1), (4, 0))),
|
|
TestInfo('(((())\n',
|
|
((0, 0), (0, 1), (1, 2), (2, 3), (3, 4), (5, 3), (6, 2))),
|
|
)
|
|
|
|
for test in tests:
|
|
with self.subTest(string=test.string):
|
|
setcode(test.string)
|
|
eq(bracketing(), test.bracket)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|