Jun 26, 2024 · Jun 26, 2024 · Jun 26, 2024 · Jun 26, 2024 · Jun 26, 2024 · Jun 26, 2024
diff --git a/Lib/test/support/__init__.py b/Lib/test/support/__init__.py
                if value is not None:
                    os.environ[key] = value
    return wrapper


 def initialized_with_pyrepl():
    """Detect whether PyREPL was used during Python initialization."""
    # If the main module has a __file__ attribute it's a Python module, which means PyREPL.
    return hasattr(sys.modules["__main__"], "__file__")
diff --git a/Lib/test/test_cmd_line.py b/Lib/test/test_cmd_line.py
        self.assertIn(expected.encode(), out)

    def test_python_basic_repl(self):
        # Currently this only tests that the env var is set
        # Currently this only tests that the env var is set. See test_pyrepl.test_python_basic_repl.
        code = "import os; print('PYTHON_BASIC_REPL' in os.environ)"
        expected = "True"
        rc, out, err = assert_python_ok('-c', code, PYTHON_BASIC_REPL='1')
diff --git a/Lib/test/test_pyrepl/test_pyrepl.py b/Lib/test/test_pyrepl/test_pyrepl.py
        self.assertNotIn("Exception", output)
        self.assertNotIn("Traceback", output)

    @force_not_colorized
    def test_python_basic_repl(self):
        env = os.environ.copy()
        commands = ("from test.support import initialized_with_pyrepl\n"
                    "initialized_with_pyrepl()\n"
                    "exit()\n")

        env.pop("PYTHON_BASIC_REPL", None)
        output, exit_code = self.run_repl(commands, env=env)
        if "can\'t use pyrepl" in output:
            self.skipTest("pyrepl not available")
        self.assertEqual(exit_code, 0)
        self.assertIn("True", output)
        self.assertNotIn("False", output)
        self.assertNotIn("Exception", output)
        self.assertNotIn("Traceback", output)

        env["PYTHON_BASIC_REPL"] = "1"
        output, exit_code = self.run_repl(commands, env=env)
        self.assertEqual(exit_code, 0)
        self.assertIn("False", output)
        self.assertNotIn("True", output)
        self.assertNotIn("Exception", output)
        self.assertNotIn("Traceback", output)

    def run_repl(self, repl_input: str | list[str], env: dict | None = None) -> tuple[str, int]:
        master_fd, slave_fd = pty.openpty()
        process = subprocess.Popen(
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2626,3 +2626,9 @@ def wrapper(args, *kwargs):
		if value is not None:
		os.environ[key] = value
		return wrapper


		def initialized_with_pyrepl():
		"""Detect whether PyREPL was used during Python initialization."""
		# If the main module has a __file__ attribute it's a Python module, which means PyREPL.
		return hasattr(sys.modules["__main__"], "__file__")
Original file line number	Diff line number	Diff line change
Expand Up		@@ -969,7 +969,7 @@ def test_python_user_base(self):
		self.assertIn(expected.encode(), out)

		def test_python_basic_repl(self):
		# Currently this only tests that the env var is set
		# Currently this only tests that the env var is set. See test_pyrepl.test_python_basic_repl.
		code = "import os; print('PYTHON_BASIC_REPL' in os.environ)"
		expected = "True"
		rc, out, err = assert_python_ok('-c', code, PYTHON_BASIC_REPL='1')
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -862,6 +862,31 @@ def test_dumb_terminal_exits_cleanly(self):
		self.assertNotIn("Exception", output)
		self.assertNotIn("Traceback", output)

		@force_not_colorized
		def test_python_basic_repl(self):
		env = os.environ.copy()
		commands = ("from test.support import initialized_with_pyrepl\n"
		"initialized_with_pyrepl()\n"
		"exit()\n")

		env.pop("PYTHON_BASIC_REPL", None)
		output, exit_code = self.run_repl(commands, env=env)
		if "can\'t use pyrepl" in output:
		self.skipTest("pyrepl not available")
		self.assertEqual(exit_code, 0)
		self.assertIn("True", output)
		self.assertNotIn("False", output)
		self.assertNotIn("Exception", output)
		self.assertNotIn("Traceback", output)

		env["PYTHON_BASIC_REPL"] = "1"
		output, exit_code = self.run_repl(commands, env=env)
		self.assertEqual(exit_code, 0)
		self.assertIn("False", output)
		self.assertNotIn("True", output)
		self.assertNotIn("Exception", output)
		self.assertNotIn("Traceback", output)

		def run_repl(self, repl_input: str \| list[str], env: dict \| None = None) -> tuple[str, int]:
		master_fd, slave_fd = pty.openpty()
		process = subprocess.Popen(
Expand Down