Clearer tests

Author: NoahStapp

test/performance/async_perf_test.py

@@ -61,7 +61,6 @@
 from bson import encode
 from gridfs import AsyncGridFSBucket
 from pymongo import (
-    AsyncMongoClient,
     DeleteOne,
     InsertOne,
     ReplaceOne,
@@ -217,14 +216,12 @@ async def do_task(self):
 class TestRunCommand8Tasks(TestRunCommand):
     n_tasks = 8
 
-    async def do_task(self):
-        command = self.client.perftest.command
-        await asyncio.gather(*[command("hello", True) for _ in range(NUM_DOCS)])
-
 
 class TestRunCommand80Tasks(TestRunCommand):
     n_tasks = 80
 
+
+class TestRunCommandManyTasks(TestRunCommand):
     async def do_task(self):
         command = self.client.perftest.command
         await asyncio.gather(*[command("hello", True) for _ in range(NUM_DOCS)])
@@ -281,14 +278,12 @@ async def do_task(self):
 class TestFindOneByID8Tasks(TestFindOneByID):
     n_tasks = 8
 
-    async def do_task(self):
-        find_one = self.corpus.find_one
-        await asyncio.gather(*[find_one({"_id": _id}) for _id in self.inserted_ids])
-
 
 class TestFindOneByID80Tasks(TestFindOneByID):
     n_tasks = 80
 
+
+class TestFindOneByIDManyTasks(TestFindOneByID):
     async def do_task(self):
         find_one = self.corpus.find_one
         await asyncio.gather(*[find_one({"_id": _id}) for _id in self.inserted_ids])
@@ -348,6 +343,10 @@ class TestFindManyAndEmptyCursor80Tasks(TestFindManyAndEmptyCursor):
     n_tasks = 80
 
 
+class TestFindManyAndEmptyCursorManyTasks(TestFindManyAndEmptyCursor):
+    n_tasks = 1000
+
+
 class TestSmallDocBulkInsert(SmallDocInsertTest, AsyncPyMongoTestCase):
     async def do_task(self):
         await self.corpus.insert_many(self.documents, ordered=True)