Skip to content

Commit 17ae80f

Browse files
author
Qian Chen
committed
modify unit-test
1 parent f9761d3 commit 17ae80f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/unit/test_handler_service.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ def __init__(self):
5252

5353
@patch('sagemaker_inference.environment.Environment')
5454
@patch('importlib.util.module_from_spec', return_value=UserModuleTransformFn())
55+
@patch('os.path.exists', return_value=True)
5556
def test_user_module_transform_fn(import_module, env):
5657
env.return_value.module_name = MODULE_NAME
5758
transformer = HandlerService._user_module_transformer()
@@ -68,6 +69,7 @@ def __init__(self):
6869

6970
@patch('sagemaker_inference.environment.Environment')
7071
@patch('importlib.util.module_from_spec', return_value=UserModuleModelFn())
72+
@patch('os.path.exists', return_value=True)
7173
def test_user_module_mxnet_module_transformer(import_module, env):
7274
env.return_value.module_name = MODULE_NAME
7375
import_module.return_value.model_fn.return_value = mx.module.BaseModule()
@@ -81,6 +83,7 @@ def test_user_module_mxnet_module_transformer(import_module, env):
8183
@patch('sagemaker_inference.environment.Environment')
8284
@patch('sagemaker_mxnet_serving_container.default_inference_handler.DefaultMXNetInferenceHandler.default_model_fn')
8385
@patch('importlib.util.module_from_spec', return_value=object())
86+
@patch('os.path.exists', return_value=True)
8487
def test_default_inference_handler_mxnet_gluon_transformer(import_module, model_fn, env):
8588
env.return_value.module_name = MODULE_NAME
8689
model_fn.return_value = mx.gluon.block.Block()

0 commit comments

Comments
 (0)