Skip to content

Commit 0e93faf

Browse files
xuanyang15copybara-github
authored andcommitted
chore: fix format
Co-authored-by: Xuan Yang <xygoogle@google.com> PiperOrigin-RevId: 886446298
1 parent 58c4536 commit 0e93faf

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

tests/unittests/tools/test_bash_tool.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -203,13 +203,16 @@ async def test_nonzero_returncode(self, workspace, tool_context_confirmed):
203203
async def test_timeout(self, workspace, tool_context_confirmed):
204204
tool = bash_tool.ExecuteBashTool(workspace=workspace)
205205
mock_process = mock.AsyncMock()
206-
with mock.patch.object(
207-
asyncio,
208-
"create_subprocess_exec",
209-
autospec=True,
210-
return_value=mock_process,
211-
), mock.patch.object(
212-
asyncio, "wait_for", autospec=True, side_effect=asyncio.TimeoutError
206+
with (
207+
mock.patch.object(
208+
asyncio,
209+
"create_subprocess_exec",
210+
autospec=True,
211+
return_value=mock_process,
212+
),
213+
mock.patch.object(
214+
asyncio, "wait_for", autospec=True, side_effect=asyncio.TimeoutError
215+
),
213216
):
214217
result = await tool.run_async(
215218
args={"command": "python scripts/do_thing.py"},

0 commit comments

Comments
 (0)