98 lines
3 KiB
Python
98 lines
3 KiB
Python
|
"""Tests for the mypy parser."""
|
||
|
|
||
|
from __future__ import annotations
|
||
|
|
||
|
import sys
|
||
|
|
||
|
from pytest import skip
|
||
|
|
||
|
from mypy import defaults
|
||
|
from mypy.config_parser import parse_mypy_comments
|
||
|
from mypy.errors import CompileError
|
||
|
from mypy.options import Options
|
||
|
from mypy.parse import parse
|
||
|
from mypy.test.data import DataDrivenTestCase, DataSuite
|
||
|
from mypy.test.helpers import assert_string_arrays_equal, find_test_files, parse_options
|
||
|
from mypy.util import get_mypy_comments
|
||
|
|
||
|
|
||
|
class ParserSuite(DataSuite):
|
||
|
required_out_section = True
|
||
|
base_path = "."
|
||
|
files = find_test_files(pattern="parse*.test", exclude=["parse-errors.test"])
|
||
|
|
||
|
if sys.version_info < (3, 10):
|
||
|
files.remove("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()
|
||
|
options.force_uppercase_builtins = True
|
||
|
options.hide_error_codes = True
|
||
|
|
||
|
if testcase.file.endswith("python310.test"):
|
||
|
options.python_version = (3, 10)
|
||
|
else:
|
||
|
options.python_version = defaults.PYTHON3_VERSION
|
||
|
|
||
|
source = "\n".join(testcase.input)
|
||
|
|
||
|
# Apply mypy: comments to options.
|
||
|
comments = get_mypy_comments(source)
|
||
|
changes, _ = parse_mypy_comments(comments, options)
|
||
|
options = options.apply_changes(changes)
|
||
|
|
||
|
try:
|
||
|
n = parse(
|
||
|
bytes(source, "ascii"), fnam="main", module="__main__", errors=None, options=options
|
||
|
)
|
||
|
a = n.str_with_options(options).split("\n")
|
||
|
except CompileError as e:
|
||
|
a = e.messages
|
||
|
assert_string_arrays_equal(
|
||
|
testcase.output, a, f"Invalid parser output ({testcase.file}, line {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})",
|
||
|
)
|