bpo-35661: Fix test for adding venv prompt to pyvenv.cfg by csabella · Pull Request #12297 · python/cpython
Expand Up
@@ -110,18 +110,20 @@ def test_defaults(self):
def test_prompt(self):
env_name = os.path.split(self.env_dir)[1]
rmtree(self.env_dir) builder = venv.EnvBuilder() self.run_with_capture(builder.create, self.env_dir) context = builder.ensure_directories(self.env_dir) self.assertEqual(context.prompt, '(%s) ' % env_name) builder.create(self.env_dir) data = self.get_text_file_contents('pyvenv.cfg') self.assertEqual(context.prompt, '(%s) ' % env_name) self.assertNotIn("prompt = ", data)
rmtree(self.env_dir) builder = venv.EnvBuilder(prompt='My prompt') self.run_with_capture(builder.create, self.env_dir) context = builder.ensure_directories(self.env_dir) self.assertEqual(context.prompt, '(My prompt) ') builder.create(self.env_dir) data = self.get_text_file_contents('pyvenv.cfg') self.assertEqual(context.prompt, '(My prompt) ') self.assertIn("prompt = 'My prompt'\n", data)
@skipInVenv Expand Down
rmtree(self.env_dir) builder = venv.EnvBuilder() self.run_with_capture(builder.create, self.env_dir) context = builder.ensure_directories(self.env_dir) self.assertEqual(context.prompt, '(%s) ' % env_name) builder.create(self.env_dir) data = self.get_text_file_contents('pyvenv.cfg') self.assertEqual(context.prompt, '(%s) ' % env_name) self.assertNotIn("prompt = ", data)
rmtree(self.env_dir) builder = venv.EnvBuilder(prompt='My prompt') self.run_with_capture(builder.create, self.env_dir) context = builder.ensure_directories(self.env_dir) self.assertEqual(context.prompt, '(My prompt) ') builder.create(self.env_dir) data = self.get_text_file_contents('pyvenv.cfg') self.assertEqual(context.prompt, '(My prompt) ') self.assertIn("prompt = 'My prompt'\n", data)
@skipInVenv Expand Down