Skip to content
Permalink
Browse files

update tests

  • Loading branch information...
yubozhao committed Jul 31, 2019
1 parent b0cf2c4 commit dcf0860a2ced9933a931c40ecd5d8e381d3da92c
Showing with 3 additions and 6 deletions.
  1. +2 −5 tests/test_service.py
  2. +1 −1 tests/test_service_env.py
@@ -41,8 +41,5 @@ def test(self, image):
empy_artifact = []
service = TestModel.pack(artifact=empy_artifact)

# ordereddict
env = service._env._conda_env._conda_env
pip_requirements = env.get('dependencies')[2]['pip']
assert pip_requirements[1] == 'imageio'
assert pip_requirements[2] == 'fastai'
assert 'imageio' in service._env._pip_dependencies
assert 'fastai' in service._env._pip_dependencies
@@ -55,7 +55,7 @@ def predict(self, df):
return df

service_with_list = ServiceWithList.pack(model=model)
saved_path = service_with_list.save(tmpdir)
saved_path = service_with_list.save(str(tmpdir))

requirements_txt_path = os.path.join(saved_path, 'requirements.txt')
with open(requirements_txt_path, 'rb') as f:

0 comments on commit dcf0860

Please sign in to comment.
You can’t perform that action at this time.