docker_test.py

# pylint: disable=wrong-or-nonexistent-copyright-notice
import subprocess
import pathlib
import platform
import pytest


def test_docker_stable():
    if platform.system() != 'Linux':
        pytest.skip("Unsupported os")
    root_folder = pathlib.Path(__file__).parent.parent
    buildResult = subprocess.run(
        ['docker', 'build', '--target', 'cirq_stable', '-t', 'cirq_image', '.'], cwd=root_folder
    )
    assert buildResult.returncode == 0

    result = subprocess.run(
        ['docker run cirq_image python -c "import cirq; assert cirq.__version__ is not None"'],
        cwd=root_folder,
        shell=True,
    )
    assert result.returncode == 0


def test_docker_pre():
    if platform.system() != 'Linux':
        pytest.skip("Unsupported os")
    root_folder = pathlib.Path(__file__).parent.parent
    buildResult = subprocess.run(
        ['docker', 'build', '--target', 'cirq_pre_release', '-t', 'cirq_image_pre', '.'],
        cwd=root_folder,
    )
    assert buildResult.returncode == 0

    result = subprocess.run(
        ['docker run cirq_image_pre python -c "import cirq; assert cirq.__version__ is not None"'],
        cwd=root_folder,
        shell=True,
    )
    assert result.returncode == 0