1 files changed, 13 insertions, 7 deletions
diff --git a/Lib/test/test_dbm.py b/Lib/test/test_dbm.py
index 4c97a4c..1a40ef6 100755
--- a/Lib/test/test_dbm.py
+++ b/Lib/test/test_dbm.py
@@ -44,12 +44,18 @@ def test_modes():
     d = dbm.open(filename, 'n')
     d.close()
 
-cleanup()
-try:
-    test_keys()
-    test_modes()
-except:
+def test_main():
     cleanup()
-    raise
+    try:
+        test_keys()
+        test_modes()
+    except:
+        cleanup()
+        raise
 
-cleanup()
+    cleanup()
+
+
+
+if __name__ == '__main__':
+    test_main()