]> granicus.if.org Git - python/commitdiff
simpler temp dir cleanup
authorSkip Montanaro <skip@pobox.com>
Sat, 17 May 2003 02:54:11 +0000 (02:54 +0000)
committerSkip Montanaro <skip@pobox.com>
Sat, 17 May 2003 02:54:11 +0000 (02:54 +0000)
Lib/test/test_bsddb185.py

index ceacf4dddc577e123dec03e54c3017e713212477..df1ece09c917936fbcddf12baaea21973dcfedda 100644 (file)
@@ -11,6 +11,7 @@ import anydbm
 import whichdb
 import os
 import tempfile
+import shutil
 
 class Bsddb185Tests(unittest.TestCase):
 
@@ -28,15 +29,12 @@ class Bsddb185Tests(unittest.TestCase):
         # Verify that anydbm.open does *not* create a bsddb185 file
         tmpdir = tempfile.mkdtemp()
         try:
-            try:
-                dbfile = os.path.join(tmpdir, "foo.db")
-                anydbm.open(os.path.splitext(dbfile)[0], "c").close()
-                ftype = whichdb.whichdb(dbfile)
-                self.assertNotEqual(ftype, "bsddb185")
-            finally:
-                os.unlink(dbfile)
+            dbfile = os.path.join(tmpdir, "foo.db")
+            anydbm.open(dbfile, "c").close()
+            ftype = whichdb.whichdb(dbfile)
+            self.assertNotEqual(ftype, "bsddb185")
         finally:
-            os.rmdir(tmpdir)
+            shutil.rmtree(tmpdir)
 
 def test_main():
     run_unittest(Bsddb185Tests)