6
6
7
7
from mypy import build
8
8
from mypy .build import BuildSource
9
- from mypy .myunit import Suite
10
9
from mypy .test .helpers import (
11
10
assert_string_arrays_equal , testfile_pyversion , normalize_error_messages
12
11
)
13
- from mypy .test .data import parse_test_cases , DataDrivenTestCase
12
+ from mypy .test .data import parse_test_cases , DataDrivenTestCase , DataSuite
14
13
from mypy .test .config import test_data_prefix , test_temp_dir
15
14
from mypy .errors import CompileError
16
- from mypy .nodes import TypeInfo
17
15
from mypy .treetransform import TransformVisitor
18
16
from mypy .types import Type
19
17
from mypy .options import Options
20
18
21
19
22
- class TransformSuite (Suite ):
20
+ class TransformSuite (DataSuite ):
23
21
# Reuse semantic analysis test cases.
24
22
transform_files = ['semanal-basic.test' ,
25
23
'semanal-expressions.test' ,
@@ -30,15 +28,19 @@ class TransformSuite(Suite):
30
28
'semanal-abstractclasses.test' ,
31
29
'semanal-python2.test' ]
32
30
33
- def cases (self ) -> List [DataDrivenTestCase ]:
31
+ @classmethod
32
+ def cases (cls ) -> List [DataDrivenTestCase ]:
34
33
c = [] # type: List[DataDrivenTestCase]
35
- for f in self .transform_files :
34
+ for f in cls .transform_files :
36
35
c += parse_test_cases (os .path .join (test_data_prefix , f ),
37
36
test_transform ,
38
37
base_path = test_temp_dir ,
39
38
native_sep = True )
40
39
return c
41
40
41
+ def run_case (self , testcase : DataDrivenTestCase ) -> None :
42
+ test_transform (testcase )
43
+
42
44
43
45
def test_transform (testcase : DataDrivenTestCase ) -> None :
44
46
"""Perform an identity transform test case."""
0 commit comments