Function test_image_with_gpu

dev/archery/archery/docker/tests/test_docker.py:546–560

(arrow_compose_path)

def test_image_with_gpu(arrow_compose_path):
    compose = DockerCompose(arrow_compose_path)

    expected_calls = [
        [
            "run", "--rm", "--gpus", "all",
            "-e", "CUDA_ENV=1",
            "-e", "OTHER_ENV=2",
            "-v", "/host:/container",
            "apache/arrow:ubuntu-cuda",
            "/bin/bash", "-c", "echo 1 > /tmp/dummy && cat /tmp/dummy",
        ]
    ]
    with assert_docker_calls(compose, expected_calls):
        compose.run('ubuntu-cuda')


def test_listing_images(arrow_compose_path):