autocmake/test/test.py

import os
import sys
import subprocess
import shlex
import shutil
import sys
import time
import datetime
import pytest

HERE = os.path.abspath(os.path.dirname(__file__))

skip_on_windows = pytest.mark.skipif('sys.platform == "win32"', reason="not working on windows")
skip_on_osx = pytest.mark.skipif('sys.platform == "darwin"', reason="not working on osx")
skip_on_linux = pytest.mark.skipif('sys.platform == "linux2"', reason="not working on linux")
skip_always = pytest.mark.skipif('1 == 1', reason="tests are broken")


# ------------------------------------------------------------------------------


def exe(command):
    """
    Executes command and returns string representations of stdout and stderr captured from the console.
    When universal_newlines=True stdout and stderr are opened in text mode.
    Otherwise, they are opened in binary mode. In that case captured stdout and stderr
    are not strings and Python 3 throws type error when compared against strings later in tests.
    Note:
    This feature is only available if Python is built with universal newline support (the default).
    Also, the newlines attribute of the file objects stdout, stdin and stderr are not updated by the
    communicate() method.
    See https://docs.python.org/2/library/subprocess.html
    """
    stdout, stderr = subprocess.Popen(shlex.split(command),
                                      stdout=subprocess.PIPE,
                                      stderr=subprocess.PIPE,
                                      universal_newlines=True).communicate()

    if stderr:
        sys.stderr.write(stdout)
        sys.stderr.write(stderr)

    return stdout, stderr

# ------------------------------------------------------------------------------


def configure_build_and_exe(name, setup_command, launcher=None):

    stamp = datetime.datetime.fromtimestamp(time.time()).strftime('%Y-%m-%d-%H-%M-%S')

    os.chdir(os.path.join(HERE, name, 'cmake'))
    shutil.copy(os.path.join('..', '..', '..', 'update.py'), 'update.py')

    dst_dir = 'lib'
    if not os.path.exists(dst_dir):
        os.makedirs(dst_dir)
    shutil.copy(os.path.join('..', '..', '..', dst_dir, 'config.py'), dst_dir)

    dst_dir = os.path.join('lib', 'docopt')
    if not os.path.exists(dst_dir):
        os.makedirs(dst_dir)
    shutil.copy(os.path.join('..', '..', '..', dst_dir, 'docopt.py'), dst_dir)

    stdout, stderr = exe('python update.py ..')
    os.chdir(os.path.join(HERE, name))

    make_command = 'make'
    binary = './bin/example'
    if sys.platform == 'win32':
        setup_command += ' --generator="MinGW Makefiles"'
        make_command = 'mingw32-make'
        binary = 'bin\\\example.exe'

    if launcher:
        binary = '%s %s' % (launcher, binary)

    setup_command += ' build-%s' % stamp

    stdout, stderr = exe(setup_command)
    assert stderr == ''

    os.chdir(os.path.join(HERE, name, 'build-%s' % stamp))

    stdout, stderr = exe(make_command)
    # we do not check for empty stderr due to warnings flushed to stderr

    stdout, stderr = exe(binary)
    assert stderr == ''

    assert 'PASSED' in stdout

# ------------------------------------------------------------------------------


def test_extra_cmake_options():
    configure_build_and_exe('extra_cmake_options', 'python setup --cxx=g++ --cmake-options="-DENABLE_SOMETHING=OFF -DENABLE_FOO=ON"')


def test_cxx():
    configure_build_and_exe('cxx', 'python setup --cxx=g++')


def test_fc():
    configure_build_and_exe('fc', 'python setup --fc=gfortran')


def test_fc_git_info():
    configure_build_and_exe('fc_git_info', 'python setup --fc=gfortran')


def test_fc_int64():
    configure_build_and_exe('fc_int64', 'python setup --fc=gfortran --int64')


def test_fc_mpi():
    if sys.platform == 'win32':
        setup_line = 'python setup --mpi --fc=gfortran --extra-fc-flags="-D_WIN64 -D INT_PTR_KIND()=8 -fno-range-check"'
    else:
        setup_line = 'python setup --mpi --fc=mpif90'
    configure_build_and_exe('fc_mpi', setup_line, 'mpiexec -np 2')


@skip_on_osx
def test_fc_omp():
    os.environ['OMP_NUM_THREADS'] = '2'
    configure_build_and_exe('fc_omp', 'python setup --omp --fc=gfortran')


def test_fc_blas():
    configure_build_and_exe('fc_blas', 'python setup --fc=gfortran --blas')


def test_fc_lapack():
    configure_build_and_exe('fc_lapack', 'python setup --fc=gfortran --lapack')


@skip_on_osx
def test_cxx_cblas():
    configure_build_and_exe('cxx_cblas', 'python setup --cxx=g++ --cblas')


@skip_on_osx
@skip_on_linux
def test_cxx_lapacke():
    configure_build_and_exe('cxx_lapacke', 'python setup --cxx=g++ --lapacke --cblas')


@skip_on_linux
@skip_on_windows
def test_cxx_accelerate():
    configure_build_and_exe('cxx_accelerate', 'python setup --cxx=g++ --accelerate')


def test_python_interpreter():
    configure_build_and_exe('python_interpreter', 'python setup --cxx=g++')


def test_python_interpreter_custom():
    setup = 'python setup --cxx=g++ --python=%s' % sys.executable
    configure_build_and_exe('python_interpreter_custom', setup)


def test_python_libs():
    configure_build_and_exe('python_libs', 'python setup --cxx=g++')


def test_python_libs_custom():
    python_executable = sys.executable
    configure_build_and_exe('python_libs_custom', 'python setup --cxx=g++ --python={}'.format(python_executable))


def test_boost_header_only():
    configure_build_and_exe('boost_header_only', 'python setup --cxx=g++')


def test_boost_libs():
    configure_build_and_exe('boost_libs', 'python setup --cxx=g++')


def test_boost_mpi_libs():
    configure_build_and_exe('boost_mpi_libs', 'python setup --cxx=g++ --mpi')


def test_boost_python_libs():
    configure_build_and_exe('boost_python_libs', 'python setup --cxx=g++')