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