Skip to content

Commit a83664b

Browse files
author
Bryannah Hernandez
committed
more unit tests
1 parent 6d023b5 commit a83664b

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

tests/unit/sagemaker/serve/mode/test_in_process_mode.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,3 +164,26 @@ def test_create_server_ex(
164164
in_process_mode._start_serving = mock_start_serving
165165

166166
self.assertRaises(InProcessDeepPingException, in_process_mode.create_server, mock_predictor)
167+
168+
@patch(
169+
"sagemaker.serve.model_server.multi_model_server.server.InProcessMultiModelServer._stop_serving"
170+
)
171+
@patch("sagemaker.serve.spec.inference_spec.InferenceSpec")
172+
@patch("sagemaker.session.Session")
173+
def test_destroy_server(
174+
self,
175+
mock_session,
176+
mock_inference_spec,
177+
mock_stop_serving,
178+
):
179+
in_process_mode = InProcessMode(
180+
model_server=ModelServer.MMS,
181+
inference_spec=mock_inference_spec,
182+
schema_builder=SchemaBuilder(mock_sample_input, mock_sample_output),
183+
session=mock_session,
184+
model_path="model_path",
185+
)
186+
187+
in_process_mode.destroy_server()
188+
189+
mock_stop_serving.assert_called()

0 commit comments

Comments
 (0)