be explicit about myunit test files

Author: elazarg

runtests.py

@@ -195,18 +195,31 @@ def add_imports(driver: Driver) -> None:
             driver.add_python_string('import %s' % mod, 'import %s' % mod)
 
 
-PYTEST_FILES = [os.path.join('mypy', 'test', '{}.py'.format(name)) for name in [
-    'testcheck',
-    'testextensions',
-    'testdeps',
-    'testdiff',
-    'testfinegrained',
-    'testmerge',
-    'testtransform',
-    'testtypegen',
-    'testparse',
-    'testsemanal',
-]]
+PYTEST_FILES = [
+    'mypy/test/testcheck.py',
+    'mypy/test/testextensions.py',
+    'mypy/test/testdeps.py',
+    'mypy/test/testdiff.py',
+    'mypy/test/testfinegrained.py',
+    'mypy/test/testmerge.py',
+    'mypy/test/testtransform.py',
+    'mypy/test/testtypegen.py',
+    'mypy/test/testparse.py',
+    'mypy/test/testsemanal.py',
+]
+
+MYUNIT_FILES = [
+    'mypy/test/teststubgen.py',  # contains data-driven suite
+
+    'mypy/test/testargs.py',
+    'mypy/test/testgraph.py',
+    'mypy/test/testinfer.py',
+    'mypy/test/testmoduleinfo.py',
+    'mypy/test/testreports.py',
+    'mypy/test/testsolve.py',
+    'mypy/test/testsubtypes.py',
+    'mypy/test/testtypes.py'
+]
 
 
 def add_pytest(driver: Driver) -> None:
@@ -233,6 +246,7 @@ def add_myunit(driver: Driver) -> None:
             # This module has been converted to pytest; don't try to use myunit.
             pass
         else:
+            assert f in MYUNIT_FILES, f
             driver.add_python_mod('unit-test %s' % mod, 'mypy.myunit', '-m', mod,
                                   *driver.arglist, coverage=True)