fix import to work with either module name.

6564ca72 · Gregory P. Smith · 6e57c2a6 · 6564ca72
Commit 6564ca72 authored Jun 09, 2005 by Gregory P. Smith
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

Lib/bsddb/test/test_compare.py Lib/bsddb/test/test_compare.py +6 -1

No files found.
--- a/Lib/bsddb/test/test_compare.py
+++ b/Lib/bsddb/test/test_compare.py
@@ -7,7 +7,12 @@ import test_all
 from cStringIO import StringIO

 import unittest
-from bsddb3 import db
+try:
+    # For Pythons w/distutils pybsddb
+    from bsddb3 import db, dbshelve
+except ImportError:
+    # For Python 2.3
+    from bsddb import db, dbshelve

 lexical_cmp = cmp