Use temporary directory

sebastinas · sebastinas · commit 2aed6b325fdd · 2021-01-01T13:50:08.000+01:00
diff --git a/bpython/test/test_repl.py b/bpython/test/test_repl.py
@@ -1,7 +1,6 @@
 import collections
 import inspect
 import os
-import shutil
 import socket
 import sys
 import tempfile
@@ -331,15 +330,11 @@ def setUp(self):
         self.repl.config.editor = "true"
 
     def test_create_config(self):
-        tmp_dir = tempfile.mkdtemp()
-        try:
+        with tempfile.TemporaryDirectory() as tmp_dir:
             config_path = Path(tmp_dir) / "newdir" / "config"
             self.repl.config.config_path = config_path
             self.repl.edit_config()
             self.assertTrue(os.path.exists(config_path))
-        finally:
-            shutil.rmtree(tmp_dir)
-            self.assertFalse(os.path.exists(config_path))
 
 
 class TestRepl(unittest.TestCase):
