@@ -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