Fixes tests failing with 3.12

PrefectHQ · Apr 2, 2024 · 6e932c5 · 6e932c5
1 parent 19bc9c4
commit 6e932c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_client_waiter.py b/tests/test_client_waiter.py
@@ -44,7 +44,7 @@ def test_flow():
         return waiter
 
     test_flow()
-    assert mock_waiter.wait.called_once_with("JobExists")
+    mock_waiter.wait.assert_called_once_with("JobExists")
 
 
 @mock_ec2
@@ -66,4 +66,4 @@ def test_flow():
         return waiter
 
     test_flow()
-    assert mock_waiter.wait.called_once_with("instance_exists")
+    mock_waiter.wait.assert_called_once_with("instance_exists")