| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- """Test cases for inferring always defined attributes in classes."""
- from __future__ import annotations
- import os.path
- from mypy.errors import CompileError
- from mypy.test.config import test_temp_dir
- from mypy.test.data import DataDrivenTestCase
- from mypyc.test.testutil import (
- ICODE_GEN_BUILTINS,
- MypycDataSuite,
- assert_test_output,
- build_ir_for_single_file2,
- infer_ir_build_options_from_test_name,
- use_custom_builtins,
- )
- files = ["alwaysdefined.test"]
- class TestAlwaysDefined(MypycDataSuite):
- files = files
- base_path = test_temp_dir
- def run_case(self, testcase: DataDrivenTestCase) -> None:
- """Perform a runtime checking transformation test case."""
- options = infer_ir_build_options_from_test_name(testcase.name)
- if options is None:
- # Skipped test case
- return
- with use_custom_builtins(os.path.join(self.data_prefix, ICODE_GEN_BUILTINS), testcase):
- try:
- ir = build_ir_for_single_file2(testcase.input, options)
- except CompileError as e:
- actual = e.messages
- else:
- actual = []
- for cl in ir.classes:
- if cl.name.startswith("_"):
- continue
- actual.append(
- "{}: [{}]".format(cl.name, ", ".join(sorted(cl._always_initialized_attrs)))
- )
- assert_test_output(testcase, actual, "Invalid test output", testcase.output)
|
