fix minor styling issue in tests

This commit is contained in:
Bastien Gérard 2019-02-25 22:33:36 +01:00
parent 3cdb5b5db2
commit c60c2ee8d0
13 changed files with 36 additions and 31 deletions

View File

@ -863,5 +863,6 @@ class DeltaTest(unittest.TestCase):
self.assertEqual('oops', delta[0]["users.007.rolist"][0]["type"]) self.assertEqual('oops', delta[0]["users.007.rolist"][0]["type"])
self.assertEqual(uinfo.id, delta[0]["users.007.info"]) self.assertEqual(uinfo.id, delta[0]["users.007.info"])
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -578,5 +578,6 @@ class FileTest(MongoDBTestCase):
self.assertEqual(marmot.photos[0].foo, 'bar') self.assertEqual(marmot.photos[0].foo, 'bar')
self.assertEqual(marmot.photos[0].get().length, 8313) self.assertEqual(marmot.photos[0].get().length, 8313)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -48,6 +48,7 @@ class PickleSignalsTest(Document):
def post_delete(self, sender, document, **kwargs): def post_delete(self, sender, document, **kwargs):
pickled = pickle.dumps(document) pickled = pickle.dumps(document)
signals.post_save.connect(PickleSignalsTest.post_save, sender=PickleSignalsTest) signals.post_save.connect(PickleSignalsTest.post_save, sender=PickleSignalsTest)
signals.post_delete.connect(PickleSignalsTest.post_delete, sender=PickleSignalsTest) signals.post_delete.connect(PickleSignalsTest.post_delete, sender=PickleSignalsTest)

View File

@ -413,7 +413,6 @@ class OnlyExcludeAllTest(unittest.TestCase):
numbers = Numbers.objects.fields(embedded__n={"$slice": [-5, 10]}).get() numbers = Numbers.objects.fields(embedded__n={"$slice": [-5, 10]}).get()
self.assertEqual(numbers.embedded.n, [-5, -4, -3, -2, -1]) self.assertEqual(numbers.embedded.n, [-5, -4, -3, -2, -1])
def test_exclude_from_subclasses_docs(self): def test_exclude_from_subclasses_docs(self):
class Base(Document): class Base(Document):
@ -436,5 +435,6 @@ class OnlyExcludeAllTest(unittest.TestCase):
self.assertRaises(LookUpError, Base.objects.exclude, "made_up") self.assertRaises(LookUpError, Base.objects.exclude, "made_up")
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -6,10 +6,12 @@ from mongoengine.connection import connect
__author__ = 'stas' __author__ = 'stas'
class Person(Document): class Person(Document):
name = StringField() name = StringField()
age = IntField() age = IntField()
class TestQuerysetPickable(unittest.TestCase): class TestQuerysetPickable(unittest.TestCase):
""" """
Test for adding pickling support for QuerySet instances Test for adding pickling support for QuerySet instances
@ -27,7 +29,6 @@ class TestQuerysetPickable(unittest.TestCase):
age=21 age=21
) )
def test_picke_simple_qs(self): def test_picke_simple_qs(self):
qs = Person.objects.all() qs = Person.objects.all()

View File

@ -275,7 +275,6 @@ class QTest(unittest.TestCase):
with self.assertRaises(InvalidQueryError): with self.assertRaises(InvalidQueryError):
self.Person.objects.filter('user1') self.Person.objects.filter('user1')
def test_q_regex(self): def test_q_regex(self):
"""Ensure that Q objects can be queried using regexes. """Ensure that Q objects can be queried using regexes.
""" """

View File

@ -302,5 +302,6 @@ class ContextManagersTest(unittest.TestCase):
_ = db.system.indexes.find_one() # queries on db.system.indexes are ignored as well _ = db.system.indexes.find_one() # queries on db.system.indexes are ignored as well
self.assertEqual(q, 1) self.assertEqual(q, 1)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -47,5 +47,6 @@ class ConnectionTest(unittest.TestCase):
self.assertEqual(conn.read_preference, READ_PREF) self.assertEqual(conn.read_preference, READ_PREF)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -39,7 +39,6 @@ class SignalTests(unittest.TestCase):
def post_init(cls, sender, document, **kwargs): def post_init(cls, sender, document, **kwargs):
signal_output.append('post_init signal, %s, document._created = %s' % (document, document._created)) signal_output.append('post_init signal, %s, document._created = %s' % (document, document._created))
@classmethod @classmethod
def pre_save(cls, sender, document, **kwargs): def pre_save(cls, sender, document, **kwargs):
signal_output.append('pre_save signal, %s' % document) signal_output.append('pre_save signal, %s' % document)
@ -431,5 +430,6 @@ class SignalTests(unittest.TestCase):
{} {}
]) ])
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()