python · encukou · Sep 13, 2024 · Jan 20, 2024 · Feb 15, 2024 · Feb 15, 2024
diff --git a/Lib/test/_test_embed_structseq.py b/Lib/test/_test_embed_structseq.py
@@ -48,7 +48,14 @@ def test_sys_funcs(self):


 try:
-    unittest.main()
+    unittest.main(
+        module=(
+            '__main__'
+            if __name__ == '__main__'
+            # Avoiding a circular import:
+            else sys.modules['test._test_embed_structseq']
+        )
+    )
 except SystemExit as exc:
    if exc.args[0] != 0:
        raise

diff --git a/Lib/test/reperf.py b/Lib/test/reperf.py
diff --git a/Lib/test/test_embed.py b/Lib/test/test_embed.py
@@ -1971,7 +1971,11 @@ def test_no_memleak(self):
    @unittest.skipUnless(support.Py_DEBUG,
                         '-X presite requires a Python debug build')
    def test_presite(self):
-        cmd = [sys.executable, "-I", "-X", "presite=test.reperf", "-c", "print('cmd')"]
+        cmd = [
+            sys.executable,
+            "-I", "-X", "presite=test._test_embed_structseq",
+            "-c", "print('unique-python-message')",
+        ]
        proc = subprocess.run(
            cmd,
            stdout=subprocess.PIPE,
@@ -1980,9 +1984,8 @@ def test_presite(self):
        )
        self.assertEqual(proc.returncode, 0)
        out = proc.stdout.strip()
-        self.assertIn("10 times sub", out)
-        self.assertIn("CPU seconds", out)
-        self.assertIn("cmd", out)
+        self.assertIn("Tests passed", out)
+        self.assertIn("unique-python-message", out)


 class StdPrinterTests(EmbeddingTestsMixin, unittest.TestCase):