87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
"""Tests for the mypy parser."""
|
|
|
|
import sys
|
|
|
|
from pytest import skip
|
|
|
|
from mypy import defaults
|
|
from mypy.test.helpers import assert_string_arrays_equal, parse_options
|
|
from mypy.test.data import DataDrivenTestCase, DataSuite
|
|
from mypy.parse import parse
|
|
from mypy.errors import CompileError
|
|
from mypy.options import Options
|
|
|
|
|
|
class ParserSuite(DataSuite):
|
|
required_out_section = True
|
|
base_path = '.'
|
|
files = ['parse.test',
|
|
'parse-python2.test']
|
|
|
|
if sys.version_info >= (3, 10):
|
|
files.append('parse-python310.test')
|
|
|
|
def run_case(self, testcase: DataDrivenTestCase) -> None:
|
|
test_parser(testcase)
|
|
|
|
|
|
def test_parser(testcase: DataDrivenTestCase) -> None:
|
|
"""Perform a single parser test case.
|
|
|
|
The argument contains the description of the test case.
|
|
"""
|
|
options = Options()
|
|
|
|
if testcase.file.endswith('python2.test'):
|
|
options.python_version = defaults.PYTHON2_VERSION
|
|
elif testcase.file.endswith('python310.test'):
|
|
options.python_version = (3, 10)
|
|
else:
|
|
options.python_version = defaults.PYTHON3_VERSION
|
|
|
|
try:
|
|
n = parse(bytes('\n'.join(testcase.input), 'ascii'),
|
|
fnam='main',
|
|
module='__main__',
|
|
errors=None,
|
|
options=options)
|
|
a = str(n).split('\n')
|
|
except CompileError as e:
|
|
a = e.messages
|
|
assert_string_arrays_equal(testcase.output, a,
|
|
'Invalid parser output ({}, line {})'.format(
|
|
testcase.file, testcase.line))
|
|
|
|
|
|
# The file name shown in test case output. This is displayed in error
|
|
# messages, and must match the file name in the test case descriptions.
|
|
INPUT_FILE_NAME = 'file'
|
|
|
|
|
|
class ParseErrorSuite(DataSuite):
|
|
required_out_section = True
|
|
base_path = '.'
|
|
files = ['parse-errors.test']
|
|
|
|
def run_case(self, testcase: DataDrivenTestCase) -> None:
|
|
test_parse_error(testcase)
|
|
|
|
|
|
def test_parse_error(testcase: DataDrivenTestCase) -> None:
|
|
try:
|
|
options = parse_options('\n'.join(testcase.input), testcase, 0)
|
|
if options.python_version != sys.version_info[:2]:
|
|
skip()
|
|
# Compile temporary file. The test file contains non-ASCII characters.
|
|
parse(bytes('\n'.join(testcase.input), 'utf-8'), INPUT_FILE_NAME, '__main__', None,
|
|
options)
|
|
raise AssertionError('No errors reported')
|
|
except CompileError as e:
|
|
if e.module_with_blocker is not None:
|
|
assert e.module_with_blocker == '__main__'
|
|
# Verify that there was a compile error and that the error messages
|
|
# are equivalent.
|
|
assert_string_arrays_equal(
|
|
testcase.output, e.messages,
|
|
f'Invalid compiler output ({testcase.file}, line {testcase.line})')
|