Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[BEAM-11797] Fix wrong assertion usage (apache#14377)
1. Replaced all assert_called_with with assert_any_call in the ProgressIndicatorTest. 2. assert_called_with only asserts the most recent call. However, there is an IPythonLogHandler we have customized to emit rich logging in notebooks that also calls display to render HTML and JavaScript. The emission of logs can happen any time during the test, causing the flakiness. Instead of unregistering the log handler during tests that could make the tests unnecessarily complicated, we could use assert_any_call within the patched context for simplicity.
- Loading branch information