Closes #22829: Added --prompt option to venv.

1 files changed, 11 insertions, 0 deletions

diff --git a/Lib/test/test_venv.py b/Lib/test/test_venv.py
index a2842b0217..f4ad7c7c5c 100644
--- a/Lib/test/test_venv.py
+++ b/Lib/test/test_venv.py
@@ -114,6 +114,17 @@ class BasicTest(BaseTest):
             print('    %r' % os.listdir(bd))
         self.assertTrue(os.path.exists(fn), 'File %r should exist.' % fn)
 
+    def test_prompt(self):
+        env_name = os.path.split(self.env_dir)[1]
+
+        builder = venv.EnvBuilder()
+        context = builder.ensure_directories(self.env_dir)
+        self.assertEqual(context.prompt, '(%s) ' % env_name)
+
+        builder = venv.EnvBuilder(prompt='My prompt')
+        context = builder.ensure_directories(self.env_dir)
+        self.assertEqual(context.prompt, '(My prompt) ')
+
     @skipInVenv
     def test_prefixes(self):
         """