Improved update in test case for removing inheritance

KarimAllah · Aug 18, 2011 · bda716e · bda716e
1 parent 91a0e49
commit bda716e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/document.py b/tests/document.py
@@ -431,7 +431,7 @@ class Animal(Document):
                 'allow_inheritance': False,
                 'indexes': ['name']
             }
-        collection.update({}, {"$unset": {"_types": 1, "_cls": 1}}, False, True)
+        collection.update({}, {"$unset": {"_types": 1, "_cls": 1}},  multi=True)
 
         # Confirm extra data is removed
         obj = collection.find_one()
@@ -1948,14 +1948,14 @@ class BlogPost(Document):
     def test_list_search_by_embedded(self):
         class User(Document):
             username = StringField(required=True)
-    
+
             meta = {'allow_inheritance': False}
-    
+
         class Comment(EmbeddedDocument):
             comment = StringField()
             user = ReferenceField(User,
                                   required=True)
-                          
+
             meta = {'allow_inheritance': False}
 
         class Page(Document):