Updates to imports for future pymongo 2.2

2012-02-17 11:18:25 +00:00
parent a15352a4f8
commit a59b518cf2
9 changed files with 81 additions and 69 deletions
--- a/tests/document.py
+++ b/tests/document.py
@@ -1,5 +1,6 @@
 import pickle
 import pymongo
+import bson
 import unittest
 import warnings

@@ -2222,7 +2223,7 @@ class DocumentTest(unittest.TestCase):

        # Test laziness
        self.assertTrue(isinstance(post_obj._data['author'],
-                                   pymongo.dbref.DBRef))
+                                   bson.DBRef))
        self.assertTrue(isinstance(post_obj.author, self.Person))
        self.assertEqual(post_obj.author.name, 'Test User')