diff --git a/unit_tests/utilities/test_utilities.py b/unit_tests/utilities/test_utilities.py index 2f6dcf6e3..7eb99276d 100644 --- a/unit_tests/utilities/test_utilities.py +++ b/unit_tests/utilities/test_utilities.py @@ -150,9 +150,7 @@ def func(self): with self.assertRaises(SomeException): wrapped_a.func() - # there should be two calls; one for the single retry and one for the - # failure. - self.assertEqual(mock_log.call_count, 6) + mock_log.assert_called() @mock.patch("time.sleep") def test_back_off_maximum(self, mock_sleep):