Fixed test for AttributedDict.

svetlyak40wt · Sep 7, 2009 · e71901c · e71901c
1 parent 75deb98
commit e71901c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_document.py b/test/test_document.py
@@ -6,6 +6,8 @@
 from pymongo.connection import Connection
 from pymongo.database import Database
 from mongobongo import Document
+from mongobongo.attributed import AttributedDict
+
 
 
 def get_connection():
@@ -96,8 +98,6 @@ def test_change_attributes(self):
         self.assertEqual(dict(title = 'Just a Poem', tags = ['one', 'two']), doc.article)
 
     def test_dict_proxy(self):
-        from pymongo.document import AttributedDict
-
         d = dict(title = 'Just an Example', tags = ['test', 'python'])
         dp = AttributedDict(d)
         dp.author = dict(name = 'Alexander')