diff --git a/py/test/selenium/webdriver/common/selenium_manager_tests.py b/py/test/selenium/webdriver/common/selenium_manager_tests.py index 67239bf87a67d..9195badf0ed21 100644 --- a/py/test/selenium/webdriver/common/selenium_manager_tests.py +++ b/py/test/selenium/webdriver/common/selenium_manager_tests.py @@ -31,9 +31,10 @@ def test_gets_results(monkeypatch): expected_output = {"driver_path": "/path/to/driver"} lib_path = "selenium.webdriver.common.selenium_manager.SeleniumManager" - with mock.patch(lib_path + "._get_binary", return_value="/path/to/sm") as mock_get_binary, mock.patch( - lib_path + "._run", return_value=expected_output - ) as mock_run: + with ( + mock.patch(lib_path + "._get_binary", return_value="/path/to/sm") as mock_get_binary, + mock.patch(lib_path + "._run", return_value=expected_output) as mock_run, + ): SeleniumManager().binary_paths([]) mock_get_binary.assert_called_once()