conftest.py

import pytest
import sys
import os

# this is import for measuring coverage
assert 'pyemma' not in sys.modules


def setup_pyemma_config():
    """ set config flags for testing """
    if 'pyemma' in sys.modules:
        pyemma = sys.modules['pyemma']
        pyemma.config.show_progress_bars = False
        pyemma.config.coordinates_check_output = True
        pyemma.config.use_trajectory_lengths_cache = False


def add_np():
    # we enforce legacy string formatting of numpy arrays, because the output format changed in version 1.14,
    # leading to failing doctests.
    import numpy as np
    try:
        np.set_printoptions(legacy='1.13')
    except TypeError:
        pass


@pytest.fixture('session')
def session_fixture():
    setup_pyemma_config()
    add_np()

    # redirect tempdir to a subdir called pyemma-test-$random to clean all temporary files after testing.
    import tempfile, uuid
    org = tempfile.gettempdir()
    tempfile.tempdir = os.path.join(org, 'pyemma_test-{}'.format(uuid.uuid4()))
    print('session temporary dir:', tempfile.tempdir)
    try:
        os.mkdir(tempfile.tempdir)
    except OSError as ose:
        if 'exists'  not in ose.strerror.lower():
            raise
    yield
    import shutil
    shutil.rmtree(tempfile.tempdir, ignore_errors=True)


def pytest_collection_modifyitems(session, config, items):
    circle_node_total, circle_node_index = read_circleci_env_variables()
    deselected = []
    for item in items:
        i = hash(item.name)
        if i % circle_node_total != circle_node_index:
            deselected.append(item)
    for item in deselected:
        items.remove(item)

    config.hook.pytest_deselected(items=deselected)


def read_circleci_env_variables():
    """Read and convert CIRCLE_* environment variables"""
    circle_node_total = int(os.environ.get("CIRCLE_NODE_TOTAL", "1").strip() or "1")
    circle_node_index = int(os.environ.get("CIRCLE_NODE_INDEX", "0").strip() or "0")

    if circle_node_index >= circle_node_total:
        raise RuntimeError("CIRCLE_NODE_INDEX={} >= CIRCLE_NODE_TOTAL={}, should be less".format(circle_node_index, circle_node_total))

    return circle_node_total, circle_node_index


def pytest_report_header(config):
    """Add CircleCI information to report"""
    circle_node_total, circle_node_index = read_circleci_env_variables()
    return "CircleCI total nodes: {}, this node index: {}".format(circle_node_total, circle_node_index)