fix test

Author: blink1073

test/asynchronous/test_transactions.py

@@ -375,6 +375,9 @@ async def test_transaction_direct_connection(self):
         async def find(*args, **kwargs):
             return coll.find(*args, **kwargs)
 
+        async def find_raw_batches(*args, **kwargs):
+            return coll.find_raw_batches(*args, **kwargs)
+
         ops = [
             (coll.bulk_write, [[InsertOne[dict]({})]]),
             (coll.insert_one, [{}]),
@@ -393,7 +396,7 @@ async def find(*args, **kwargs):
             (coll.aggregate, [[]]),
             (find, [{}]),
             (coll.aggregate_raw_batches, [[]]),
-            (coll.find_raw_batches, [{}]),
+            (find_raw_batches, [{}]),
             (coll.database.command, ["find", coll.name]),
         ]
         for f, args in ops:

test/test_transactions.py

@@ -371,6 +371,9 @@ def test_transaction_direct_connection(self):
         def find(*args, **kwargs):
             return coll.find(*args, **kwargs)
 
+        def find_raw_batches(*args, **kwargs):
+            return coll.find_raw_batches(*args, **kwargs)
+
         ops = [
             (coll.bulk_write, [[InsertOne[dict]({})]]),
             (coll.insert_one, [{}]),
@@ -389,7 +392,7 @@ def find(*args, **kwargs):
             (coll.aggregate, [[]]),
             (find, [{}]),
             (coll.aggregate_raw_batches, [[]]),
-            (coll.find_raw_batches, [{}]),
+            (find_raw_batches, [{}]),
             (coll.database.command, ["find", coll.name]),
         ]
         for f, args in ops: