modify test

Author: Qian Chen

test/unit/test_handler_service.py

@@ -53,7 +53,7 @@ def __init__(self):
 @patch('sagemaker_inference.environment.Environment')
 @patch('importlib.util.module_from_spec', return_value=UserModuleTransformFn())
 @patch('os.path.exists', return_value=True)
-def test_user_module_transform_fn(import_module, env):
+def test_user_module_transform_fn(import_module, env, path_exists):
     env.return_value.module_name = MODULE_NAME
     transformer = HandlerService._user_module_transformer()
 
@@ -70,7 +70,7 @@ def __init__(self):
 @patch('sagemaker_inference.environment.Environment')
 @patch('importlib.util.module_from_spec', return_value=UserModuleModelFn())
 @patch('os.path.exists', return_value=True)
-def test_user_module_mxnet_module_transformer(import_module, env):
+def test_user_module_mxnet_module_transformer(import_module, env, path_exists):
     env.return_value.module_name = MODULE_NAME
     import_module.return_value.model_fn.return_value = mx.module.BaseModule()
 
@@ -84,7 +84,7 @@ def test_user_module_mxnet_module_transformer(import_module, env):
 @patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultMXNetInferenceHandler.default_model_fn')
 @patch('importlib.util.module_from_spec', return_value=object())
 @patch('os.path.exists', return_value=True)
-def test_default_inference_handler_mxnet_gluon_transformer(import_module, model_fn, env):
+def test_default_inference_handler_mxnet_gluon_transformer(import_module, model_fn, env, path_exists):
     env.return_value.module_name = MODULE_NAME
     model_fn.return_value = mx.gluon.block.Block()