Ejemplos de download_and_install en Python

Ejemplo n.º 1

Archivo: test_modules.py Proyecto: wiltonwu/sagemaker-containers

def test_download_and_install_local_directory(install, prepare, tarfile, s3_download):
    uri = "/opt/ml/input/data/code/sourcedir.tar.gz"
    _modules.download_and_install(uri)

    s3_download.assert_not_called()
    tarfile.assert_called_with(name="/opt/ml/input/data/code/sourcedir.tar.gz", mode="r:gz")
    prepare.assert_called_once()
    install.assert_called_once()

Ejemplo n.º 2

Archivo: test_modules.py Proyecto: wiltonwu/sagemaker-containers

def test_download_and_install(install, prepare, download_and_extract, files_tmpdir, module_exists):
    files_tmpdir.return_value.__enter__.return_value = "tmp"
    uri = "s3://foo/bar"
    _modules.download_and_install(uri)

    module_path = os.path.join("tmp", "module_dir")
    download_and_extract.assert_called_with(uri, module_path)
    prepare.assert_called_with(module_path, "default_user_module_name")
    install.assert_called_with(module_path)

Ejemplo n.º 3

Archivo: test_modules.py Proyecto: bearpelican/sagemaker-containers

def test_download_and_install_local_directory():
    uri = '/opt/ml/code'

    with patch('sagemaker_containers._modules.s3_download') as s3_download, \
            patch('sagemaker_containers._modules.prepare') as prepare, \
            patch('sagemaker_containers._modules.install') as install:
        _modules.download_and_install(uri)

        s3_download.assert_not_called()
        prepare.assert_called_with(uri, 'default_user_module_name')
        install.assert_called_with(uri)

Ejemplo n.º 4

Archivo: test_modules.py Proyecto: torikoshi/sagemaker-containers

def test_download_and_install_local_directory():
    uri = "/opt/ml/code"

    with patch(
            "sagemaker_containers._files.s3_download") as s3_download, patch(
                "sagemaker_containers._modules.prepare") as prepare, patch(
                    "sagemaker_containers._modules.install") as install:
        _modules.download_and_install(uri)

        s3_download.assert_not_called()
        prepare.assert_called_with(uri, "default_user_module_name")
        install.assert_called_with(uri)

Ejemplo n.º 5

def test_download_and_install_local_directory():
    uri = "/opt/ml/input/data/code/sourcedir.tar.gz"

    with patch("sagemaker_containers._files.s3_download"
               ) as s3_download, patch("tarfile.open") as tarfile, patch(
                   "sagemaker_containers._modules.prepare") as prepare, patch(
                       "sagemaker_containers._modules.install") as install:
        _modules.download_and_install(uri)

        s3_download.assert_not_called()
        tarfile.assert_called_with(
            name="/opt/ml/input/data/code/sourcedir.tar.gz", mode="r:gz")
        prepare.assert_called_once()
        install.assert_called_once()