change unit-test

Author: Qian Chen

test/unit/test_handler_service.py

@@ -51,7 +51,7 @@ def __init__(self):
 
 
 @patch('sagemaker_inference.environment.Environment')
-@patch('importlib.import_module', return_value=UserModuleTransformFn())
+@patch('importlib.util.module_from_spec', return_value=UserModuleTransformFn())
 def test_user_module_transform_fn(import_module, env):
     env.return_value.module_name = MODULE_NAME
     transformer = HandlerService._user_module_transformer()
@@ -67,7 +67,7 @@ def __init__(self):
 
 
 @patch('sagemaker_inference.environment.Environment')
-@patch('importlib.import_module', return_value=UserModuleModelFn())
+@patch('importlib.util.module_from_spec', return_value=UserModuleModelFn())
 def test_user_module_mxnet_module_transformer(import_module, env):
     env.return_value.module_name = MODULE_NAME
     import_module.return_value.model_fn.return_value = mx.module.BaseModule()
@@ -80,7 +80,7 @@ def test_user_module_mxnet_module_transformer(import_module, env):
 
 @patch('sagemaker_inference.environment.Environment')
 @patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultMXNetInferenceHandler.default_model_fn')
-@patch('importlib.import_module', return_value=object())
+@patch('importlib.util.module_from_spec', return_value=object())
 def test_default_inference_handler_mxnet_gluon_transformer(import_module, model_fn, env):
     env.return_value.module_name = MODULE_NAME
     model_fn.return_value = mx.gluon.block.Block()
@@ -94,7 +94,7 @@ def test_default_inference_handler_mxnet_gluon_transformer(import_module, model_
 
 
 @patch('sagemaker_inference.environment.Environment')
-@patch('importlib.import_module', return_value=UserModuleModelFn())
+@patch('importlib.util.module_from_spec', return_value=UserModuleModelFn())
 def test_user_module_unsupported(import_module, env):
     env.return_value.module_name = MODULE_NAME