File tree Expand file tree Collapse file tree 1 file changed +23
-0
lines changed
tests/unit/sagemaker/serve/mode Expand file tree Collapse file tree 1 file changed +23
-0
lines changed Original file line number Diff line number Diff line change @@ -164,3 +164,26 @@ def test_create_server_ex(
164
164
in_process_mode ._start_serving = mock_start_serving
165
165
166
166
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 ()
You can’t perform that action at this time.
0 commit comments