Update Tests for EmbeddedDocument Unique Indicies

2014-11-20 11:20:04 -05:00
parent bc53dd6830
commit 43af9f3fad
3 changed files with 18 additions and 17 deletions
--- a/tests/fields/fields.py
+++ b/tests/fields/fields.py
@@ -1974,14 +1974,14 @@ class FieldTest(unittest.TestCase):
    def test_recursive_embedding(self):
        """Ensure that EmbeddedDocumentFields can contain their own documents.
        """
-        class Tree(Document):
-            name = StringField()
-            children = ListField(EmbeddedDocumentField('TreeNode'))
-
        class TreeNode(EmbeddedDocument):
            name = StringField()
            children = ListField(EmbeddedDocumentField('self'))

+        class Tree(Document):
+            name = StringField()
+            children = ListField(EmbeddedDocumentField('TreeNode'))
+
        Tree.drop_collection()
        tree = Tree(name="Tree")