From bda716ef9d721248a9f3d5501c1e3e5a2637c442 Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Thu, 18 Aug 2011 08:30:52 +0100
Subject: [PATCH] Improved update in test case for removing inheritance

---
 tests/document.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/document.py b/tests/document.py
index cfeffb75..d6d70289 100644
--- a/tests/document.py
+++ b/tests/document.py
@@ -431,7 +431,7 @@ class DocumentTest(unittest.TestCase):
                 '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 DocumentTest(unittest.TestCase):
     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):