fix poor assert's in tests

This commit is contained in:
Bastien Gérard
2019-02-25 22:29:44 +01:00
parent b9cc8a4ca9
commit 3cdb5b5db2
5 changed files with 27 additions and 23 deletions

View File

@@ -2168,15 +2168,15 @@ class FieldTest(MongoDBTestCase):
to_embed = ToEmbed(id=2, recursive=to_embed_recursive).save()
doc = Doc(field=to_embed)
doc.save()
assert isinstance(doc.field, ToEmbed)
assert doc.field == to_embed
self.assertIsInstance(doc.field, ToEmbed)
self.assertEqual(doc.field, to_embed)
# Same thing with a Document with a _cls field
to_embed_recursive = ToEmbedChild(id=1).save()
to_embed_child = ToEmbedChild(id=2, recursive=to_embed_recursive).save()
doc = Doc(field=to_embed_child)
doc.save()
assert isinstance(doc.field, ToEmbedChild)
assert doc.field == to_embed_child
self.assertIsInstance(doc.field, ToEmbedChild)
self.assertEqual(doc.field, to_embed_child)
def test_cls_field(self):
class Animal(Document):