Require output fixture extension to match sourceType output.

packages/babel-core/src/transformation/index.js

@@ -67,6 +67,7 @@ export function runSync(
     ast: opts.ast === true ? file.ast : null,
     code: outputCode === undefined ? null : outputCode,
     map: outputMap === undefined ? null : outputMap,
+    sourceType: file.ast.program.sourceType,
   };
 }
 

packages/babel-helper-transform-fixture-test-runner/src/index.js

@@ -397,11 +397,28 @@ function run(task) {
       fs.statSync(path.dirname(expected.loc)).isDirectory() &&
       !process.env.CI
     ) {
-      console.log(`New test file created: ${expected.loc}`);
+      const expectedFile = expected.loc.replace(
-      fs.writeFileSync(expected.loc, `${result.code}\n`);
+        /\.m?js$/,
+        result.sourceType === "module" ? ".mjs" : ".js",
+      );
+
+      console.log(`New test file created: ${expectedFile}`);
+      fs.writeFileSync(expectedFile, `${result.code}\n`);
+
+      if (expected.loc !== expectedFile) {
+        try {
+          fs.unlinkSync(expected.loc);
+        } catch (e) {}
+      }
     } else {
       actualCode = result.code.trim();
       expect(actualCode).toEqual(expectCode);
+
+      if (actualCode) {
+        expect(expected.loc).toMatch(
+          result.sourceType === "module" ? /\.mjs$/ : /\.js$/,
+        );
+      }
     }
   }