Skip to content

Commit e9cfbe7

Browse files
elazarggvanrossum
authored andcommitted
move testtransform to pytest (#3788)
1 parent 9488e38 commit e9cfbe7

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

mypy/test/testtransform.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,18 @@
66

77
from mypy import build
88
from mypy.build import BuildSource
9-
from mypy.myunit import Suite
109
from mypy.test.helpers import (
1110
assert_string_arrays_equal, testfile_pyversion, normalize_error_messages
1211
)
13-
from mypy.test.data import parse_test_cases, DataDrivenTestCase
12+
from mypy.test.data import parse_test_cases, DataDrivenTestCase, DataSuite
1413
from mypy.test.config import test_data_prefix, test_temp_dir
1514
from mypy.errors import CompileError
16-
from mypy.nodes import TypeInfo
1715
from mypy.treetransform import TransformVisitor
1816
from mypy.types import Type
1917
from mypy.options import Options
2018

2119

22-
class TransformSuite(Suite):
20+
class TransformSuite(DataSuite):
2321
# Reuse semantic analysis test cases.
2422
transform_files = ['semanal-basic.test',
2523
'semanal-expressions.test',
@@ -30,15 +28,19 @@ class TransformSuite(Suite):
3028
'semanal-abstractclasses.test',
3129
'semanal-python2.test']
3230

33-
def cases(self) -> List[DataDrivenTestCase]:
31+
@classmethod
32+
def cases(cls) -> List[DataDrivenTestCase]:
3433
c = [] # type: List[DataDrivenTestCase]
35-
for f in self.transform_files:
34+
for f in cls.transform_files:
3635
c += parse_test_cases(os.path.join(test_data_prefix, f),
3736
test_transform,
3837
base_path=test_temp_dir,
3938
native_sep=True)
4039
return c
4140

41+
def run_case(self, testcase: DataDrivenTestCase) -> None:
42+
test_transform(testcase)
43+
4244

4345
def test_transform(testcase: DataDrivenTestCase) -> None:
4446
"""Perform an identity transform test case."""

runtests.py

+1
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,7 @@ def add_imports(driver: Driver) -> None:
204204
'testdiff',
205205
'testfinegrained',
206206
'testmerge',
207+
'testtransform',
207208
'testparse',
208209
]]
209210

0 commit comments

Comments
 (0)