Fix test failure

pyocd · Jul 13, 2024 · 9b937c1 · 9b937c1
1 parent 456cbf8
commit 9b937c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/unit/test_autoflush.py b/test/unit/test_autoflush.py
@@ -29,11 +29,11 @@ class TestAutoflush:
     def test_flushed(self, mock_obj):
         with Autoflush(mock_obj):
             pass
-        assert mock_obj.flush.called
+        assert mock_obj.flush.assert_called
 
     def test_transfer_err_not_flushed(self, mock_obj):
         with pytest.raises(TransferError):
             with Autoflush(mock_obj):
                 raise TransferError("bad joojoo")
-        assert mock_obj.flush.not_called
+        assert mock_obj.flush.assert_not_called