def test_download_and_extract_source_dir(copy, rmtree, s3_download):
uri = _env.channel_path("code")
_files.download_and_extract(uri, _env.code_dir)
s3_download.assert_not_called()
rmtree.assert_any_call(_env.code_dir)
copy.assert_called_with(uri, _env.code_dir)
def test_download_and_and_extract_source_dir(move, rmtree, s3_download):
uri = _env.channel_path('code')
_files.download_and_extract(uri, 'train.sh', _env.code_dir)
s3_download.assert_not_called()
rmtree.assert_any_call(_env.code_dir)
move.assert_called_with(uri, _env.code_dir)
def test_channel_input_dirs():
input_data_path = _env._input_data_dir
assert _env.channel_path('evaluation') == os.path.join(
input_data_path, 'evaluation')
assert _env.channel_path('training') == os.path.join(
input_data_path, 'training')
def test_download_and_and_extract_file(copy, s3_download):
uri = _env.channel_path('code')
_files.download_and_extract(uri, _env.code_dir)
s3_download.assert_not_called()
copy.assert_called_with(uri, _env.code_dir)
def test_channel_input_dirs():
input_data_path = _env._input_data_dir
assert _env.channel_path("evaluation") == os.path.join(
input_data_path, "evaluation")
assert _env.channel_path("training") == os.path.join(
input_data_path, "training")
