test8

Author: Qian Chen

src/sagemaker_mxnet_serving_container/handler_service.py

@@ -54,7 +54,8 @@ def _user_module_transformer(model_dir=environment.model_dir):
             user_module = importlib.util.module_from_spec(spec)
         else:
             log.info("Please include /code/{}.py in model_dir".format(module_name))
-            raise ValueError("Could not find valid inference_script path {} in model artifact".format(inference_script))
+            raise ValueError('Invalid inference_script path: Could not find '
+                     'valid inference_script path {} in model artifact'.format(inference_script))
 
         if hasattr(user_module, 'transform_fn'):
             return Transformer(default_inference_handler=DefaultMXNetInferenceHandler())

test/unit/test_handler_service.py

@@ -107,3 +107,13 @@ def test_user_module_unsupported(path_exists, module_from_spec, env):
 
     module_from_spec.assert_called_once()
     e.match('Unsupported model type')
+
+@patch('sagemaker_inference.environment.Environment')
+@patch('importlib.util.module_from_spec', return_value=UserModuleModelFn())
+def test_user_module_unsupported(module_from_spec, env):
+    env.return_value.module_name = MODULE_NAME
+
+    with pytest.raises(ValueError) as e:
+        HandlerService._user_module_transformer()
+
+    e.match('Invalid inference_script path')