rename remaining files for pytest migration

tests/queryset/test_modify.py

@@ -1,8 +1,6 @@
 import unittest
 
-from mongoengine import connect, Document, IntField, StringField, ListField
+from mongoengine import Document, IntField, ListField, StringField, connect
-
-__all__ = ("FindAndModifyTest",)
 
 
 class Doc(Document):
@@ -10,7 +8,7 @@ class Doc(Document):
     value = IntField()
 
 
-class FindAndModifyTest(unittest.TestCase):
+class TestFindAndModify(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")
         Doc.drop_collection()

tests/queryset/test_pickable.py

@@ -1,10 +1,8 @@
 import pickle
 import unittest
-from pymongo.mongo_client import MongoClient
-from mongoengine import Document, StringField, IntField
-from mongoengine.connection import connect
 
-__author__ = "stas"
+from mongoengine import Document, IntField, StringField
+from mongoengine.connection import connect
 
 
 class Person(Document):
@@ -20,11 +18,8 @@ class TestQuerysetPickable(unittest.TestCase):
 
     def setUp(self):
         super(TestQuerysetPickable, self).setUp()
-
+        connection = connect(db="test")
-        connection = connect(db="test")  # type: pymongo.mongo_client.MongoClient
-
         connection.drop_database("test")
-
         self.john = Person.objects.create(name="John", age=21)
 
     def test_picke_simple_qs(self):

tests/queryset/test_transform.py

@@ -5,10 +5,8 @@ from bson.son import SON
 from mongoengine import *
 from mongoengine.queryset import Q, transform
 
-__all__ = ("TransformTest",)
 
-
+class TestTransform(unittest.TestCase):
-class TransformTest(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")
 

tests/queryset/test_visitor.py

@@ -8,10 +8,8 @@ from mongoengine import *
 from mongoengine.errors import InvalidQueryError
 from mongoengine.queryset import Q
 
-__all__ = ("QTest",)
 
-
+class TestQ(unittest.TestCase):
-class QTest(unittest.TestCase):
     def setUp(self):
         connect(db="mongoenginetest")