Fix predicates

Author: NoahStapp

test/asynchronous/test_examples.py

@@ -1409,8 +1409,14 @@ async def test_snapshot_query(self):
             {"name": "Pebbles", "color": "Brown", "age": 10, "adoptable": True}
         )
 
-        await async_wait_until(functools.partial(self.check_for_snapshot, db.cats), "success")
-        await async_wait_until(functools.partial(self.check_for_snapshot, db.dogs), "success")
+        async def predicate_one():
+            return await self.check_for_snapshot(db.cats)
+
+        async def predicate_two():
+            return await self.check_for_snapshot(db.dogs)
+
+        await async_wait_until(predicate_two, "success")
+        await async_wait_until(predicate_one, "success")
 
         # Start Snapshot Query Example 1
 
@@ -1443,7 +1449,11 @@ async def test_snapshot_query(self):
 
         saleDate = datetime.datetime.now()
         await db.sales.insert_one({"shoeType": "boot", "price": 30, "saleDate": saleDate})
-        await async_wait_until(functools.partial(self.check_for_snapshot, db.sales), "success")
+
+        async def predicate_three():
+            return await self.check_for_snapshot(db.sales)
+
+        await async_wait_until(predicate_three, "success")
 
         # Start Snapshot Query Example 2
         db = client.retail

test/test_examples.py

@@ -1403,8 +1403,14 @@ def test_snapshot_query(self):
         db.cats.insert_one({"name": "Whiskers", "color": "white", "age": 10, "adoptable": True})
         db.dogs.insert_one({"name": "Pebbles", "color": "Brown", "age": 10, "adoptable": True})
 
-        wait_until(functools.partial(self.check_for_snapshot, db.cats), "success")
-        wait_until(functools.partial(self.check_for_snapshot, db.dogs), "success")
+        def predicate_one():
+            return self.check_for_snapshot(db.cats)
+
+        def predicate_two():
+            return self.check_for_snapshot(db.dogs)
+
+        wait_until(predicate_two, "success")
+        wait_until(predicate_one, "success")
 
         # Start Snapshot Query Example 1
 
@@ -1437,7 +1443,11 @@ def test_snapshot_query(self):
 
         saleDate = datetime.datetime.now()
         db.sales.insert_one({"shoeType": "boot", "price": 30, "saleDate": saleDate})
-        wait_until(functools.partial(self.check_for_snapshot, db.sales), "success")
+
+        def predicate_three():
+            return self.check_for_snapshot(db.sales)
+
+        wait_until(predicate_three, "success")
 
         # Start Snapshot Query Example 2
         db = client.retail