Improve tests

Author: JukkaL

mypy/test/testexportjson.py

@@ -30,8 +30,11 @@ def run_case(self, testcase: DataDrivenTestCase) -> None:
             options.show_traceback = True
             options.allow_empty_bodies = True
             options.fixed_format_cache = True
+            fnam = os.path.join(self.base_path, "main.py")
+            with open(fnam, "w") as f:
+                f.write(src)
             result = build.build(
-                sources=[BuildSource("main", None, src)],
+                sources=[BuildSource(fnam, "main")],
                 options=options,
                 alt_lib_path=test_temp_dir,
             )
@@ -41,7 +44,7 @@ def run_case(self, testcase: DataDrivenTestCase) -> None:
             cache_dir = os.path.join(".mypy_cache", f"{major}.{minor}")
 
             for module in result.files:
-                if module in ("builtins", "typing", "_typeshed", "__main__"):
+                if module in ("builtins", "typing", "_typeshed"):
                     continue
                 fnam = os.path.join(cache_dir, f"{module}.data.ff")
                 with open(fnam, "rb") as f:

test-data/unit/exportjson.test

@@ -4,15 +4,11 @@
 -- full test coverage.
 
 [case testExportVar]
-import m
-
-[file m.py]
 x = 0
-
 [out]
 {
     ".class": "MypyFile",
-    "_fullname": "m",
+    "_fullname": "main",
     "names": {
         ".class": "SymbolTable",
         "x": {
@@ -21,7 +17,7 @@ x = 0
             "node": {
                 ".class": "Var",
                 "name": "x",
-                "fullname": "m.x",
+                "fullname": "main.x",
                 "type": {
                     ".class": "Instance",
                     "type_ref": null,
@@ -43,25 +39,21 @@ x = 0
 }
 
 [case testExportClass]
-import m
-
-[file m.py]
 class C:
     x: int
-
 [out]
 {
     ".class": "MypyFile",
-    "_fullname": "m",
+    "_fullname": "main",
     "names": {
         ".class": "SymbolTable",
         "C": {
             ".class": "SymbolTableNode",
             "kind": "Gdef",
             "node": {
                 ".class": "TypeInfo",
-                "module_name": "m",
-                "fullname": "m.C",
+                "module_name": "main",
+                "fullname": "main.C",
                 "names": {
                     ".class": "SymbolTable",
                     "x": {
@@ -70,7 +62,7 @@ class C:
                         "node": {
                             ".class": "Var",
                             "name": "x",
-                            "fullname": "m.C.x",
+                            "fullname": "main.C.x",
                             "type": {
                                 ".class": "Instance",
                                 "type_ref": null,
@@ -87,7 +79,7 @@ class C:
                 "defn": {
                     ".class": "ClassDef",
                     "name": "C",
-                    "fullname": "m.C",
+                    "fullname": "main.C",
                     "type_vars": []
                 },
                 "abstract_attributes": [],