From ab90f9b7b2c78cfcea6c91981814cb448aae9ae6 Mon Sep 17 00:00:00 2001
From: elazar <elazarg@gmail.com>
Date: Mon, 31 Jul 2017 01:11:21 +0300
Subject: [PATCH] move testtransform to pytest

---
 mypy/test/testtransform.py | 14 ++++++++------
 runtests.py                |  1 +
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/mypy/test/testtransform.py b/mypy/test/testtransform.py
index 0dcdd1d0c649..852b96721d40 100644
--- a/mypy/test/testtransform.py
+++ b/mypy/test/testtransform.py
@@ -6,20 +6,18 @@
 
 from mypy import build
 from mypy.build import BuildSource
-from mypy.myunit import Suite
 from mypy.test.helpers import (
     assert_string_arrays_equal, testfile_pyversion, normalize_error_messages
 )
-from mypy.test.data import parse_test_cases, DataDrivenTestCase
+from mypy.test.data import parse_test_cases, DataDrivenTestCase, DataSuite
 from mypy.test.config import test_data_prefix, test_temp_dir
 from mypy.errors import CompileError
-from mypy.nodes import TypeInfo
 from mypy.treetransform import TransformVisitor
 from mypy.types import Type
 from mypy.options import Options
 
 
-class TransformSuite(Suite):
+class TransformSuite(DataSuite):
     # Reuse semantic analysis test cases.
     transform_files = ['semanal-basic.test',
                        'semanal-expressions.test',
@@ -30,15 +28,19 @@ class TransformSuite(Suite):
                        'semanal-abstractclasses.test',
                        'semanal-python2.test']
 
-    def cases(self) -> List[DataDrivenTestCase]:
+    @classmethod
+    def cases(cls) -> List[DataDrivenTestCase]:
         c = []  # type: List[DataDrivenTestCase]
-        for f in self.transform_files:
+        for f in cls.transform_files:
             c += parse_test_cases(os.path.join(test_data_prefix, f),
                                   test_transform,
                                   base_path=test_temp_dir,
                                   native_sep=True)
         return c
 
+    def run_case(self, testcase: DataDrivenTestCase) -> None:
+        test_transform(testcase)
+
 
 def test_transform(testcase: DataDrivenTestCase) -> None:
     """Perform an identity transform test case."""
diff --git a/runtests.py b/runtests.py
index fdb824fd7864..c42899b21a7f 100755
--- a/runtests.py
+++ b/runtests.py
@@ -204,6 +204,7 @@ def add_imports(driver: Driver) -> None:
     'testdiff',
     'testfinegrained',
     'testmerge',
+    'testtransform',
     'testparse',
 ]]