Skip to content

Commit 3dc2918

Browse files
refactor: pyink fixes
1 parent 67a4b42 commit 3dc2918

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

tests/unittests/code_executors/test_agent_engine_sandbox_code_executor.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -131,14 +131,14 @@ def test_init_with_agent_engine_resource_name_and_ttl(
131131
mock_api_client = MagicMock()
132132
mock_vertexai_client.return_value = mock_api_client
133133
mock_operation = MagicMock()
134-
mock_operation.response.name = (
135-
"projects/123/locations/us-central1/reasoningEngines/456/sandboxEnvironments/789"
136-
)
134+
mock_operation.response.name = "projects/123/locations/us-central1/reasoningEngines/456/sandboxEnvironments/789"
137135
mock_api_client.agent_engines.sandboxes.create.return_value = mock_operation
138136

139137
# Execute
140138
executor = AgentEngineSandboxCodeExecutor(
141-
agent_engine_resource_name="projects/123/locations/us-central1/reasoningEngines/456",
139+
agent_engine_resource_name=(
140+
"projects/123/locations/us-central1/reasoningEngines/456"
141+
),
142142
sandbox_ttl="3600s",
143143
)
144144

@@ -164,14 +164,14 @@ def test_init_with_agent_engine_resource_name_without_ttl(
164164
mock_api_client = MagicMock()
165165
mock_vertexai_client.return_value = mock_api_client
166166
mock_operation = MagicMock()
167-
mock_operation.response.name = (
168-
"projects/123/locations/us-central1/reasoningEngines/456/sandboxEnvironments/789"
169-
)
167+
mock_operation.response.name = "projects/123/locations/us-central1/reasoningEngines/456/sandboxEnvironments/789"
170168
mock_api_client.agent_engines.sandboxes.create.return_value = mock_operation
171169

172170
# Execute
173171
executor = AgentEngineSandboxCodeExecutor(
174-
agent_engine_resource_name="projects/123/locations/us-central1/reasoningEngines/456",
172+
agent_engine_resource_name=(
173+
"projects/123/locations/us-central1/reasoningEngines/456"
174+
),
175175
)
176176

177177
# Assert
@@ -183,4 +183,3 @@ def test_init_with_agent_engine_resource_name_without_ttl(
183183
display_name="default_sandbox",
184184
ttl=None,
185185
)
186-

0 commit comments

Comments
 (0)