conftest.py

"""
Title:   conftest.py
Purpose: Contains hooks and fixtures for yt testing
Notes:
    1.) https://docs.pytest.org/en/latest/example/simple.html
    2.) https://docs.pytest.org/en/latest/historical-notes.html#string-conditions
    3.) https://docs.pytest.org/en/latest/skipping.html#skipif
    4.) https://docs.pytest.org/en/latest/reference.html
"""
import os
import shutil
import tempfile

import pytest
import yaml

import yt
from yt.config import ytcfg
from yt.utilities.answer_testing import utils

# Global variables can be added to the pytest namespace
answer_files = {}

# List of answer files
answer_file_list = "tests/tests.yaml"
answer_dir = os.path.join(ytcfg.get("yt", "test_data_dir"), "answers")


def pytest_addoption(parser):
    """
    Lets options be passed to test functions.
    """
    parser.addoption(
        "--with-answer-testing", action="store_true", default=False,
    )
    parser.addoption(
        "--answer-store", action="store_true", default=False,
    )
    parser.addoption(
        "--answer-big-data", action="store_true", default=False,
    )
    parser.addoption(
        "--save-answer-arrays", action="store_true",
    )


def pytest_configure(config):
    r"""
    Reads in the tests/tests.yaml file. This file contains a list of
    each answer test's answer file (including the changeset number).
    """
    yt._called_from_pytest = True
    # Make sure that the answers dir exists. If not, try to make it
    if not os.path.isdir(answer_dir):
        os.mkdir(answer_dir)
    # Read the list of answer test classes and their associated answer
    # file
    with open(answer_file_list, "r") as f:
        answer_files = yaml.safe_load(f)
    # Register custom marks for answer tests and big data
    config.addinivalue_line("markers", "answer_test: Run the answer tests.")
    config.addinivalue_line(
        "markers", "big_data: Run answer tests that require" " large data files."
    )


def pytest_collection_modifyitems(config, items):
    r"""
    Decide which tests to skip based on command-line options.
    """
    # Set up the skip marks
    skip_answer = pytest.mark.skip(reason="--with-answer-testing not set.")
    skip_big = pytest.mark.skip(reason="--answer-big-data not set.")
    # Loop over every collected test function
    for item in items:
        # If it's an answer test and the appropriate CL option hasn't
        # been set, skip it
        if "answer_test" in item.keywords and not config.getoption(
            "--with-answer-testing"
        ):
            item.add_marker(skip_answer)
        # If it's an answer test that requires big data and the CL
        # option hasn't been set, skip it
        if "big_data" in item.keywords and not config.getoption("--answer-big-data"):
            item.add_marker(skip_big)


@pytest.fixture(scope="function")
def temp_dir():
    r"""
    Creates a temporary directory needed by certain tests.
    """
    curdir = os.getcwd()
    if int(os.environ.get("GENERATE_YTDATA", 0)):
        tmpdir = os.getcwd()
    else:
        tmpdir = tempfile.mkdtemp()
    os.chdir(tmpdir)
    yield tmpdir
    os.chdir(curdir)
    if tmpdir != curdir:
        shutil.rmtree(tmpdir)


@pytest.fixture(scope="class")
def answer_file(request):
    r"""
    Assigns the name of the appropriate answer file as an attribute of
    the calling answer test class.

    The answer file is the file that either already contains previously
    generated answers or else is going to contain newly generated
    answers. If an answer file already exists, then it cannot be used
    to hold newly generated answers; a fresh file must be used instead.

    Each class that performs answer tests (e.g., TestEnzo) has a
    corresponding answer file. These answer files are stored in
    test_data_dir/answers. The names of the files and their
    corresponding answer class are in ./tests/tests.yaml.

    Parameters:
    -----------
        request : pytest.FixtureRequest
            Provides access to the requesting test context. For
            example, if an answer class uses this fixture, such as
            TestEnzo, then request provides access to all of the
            methods and attributes of the TestEnzo class, since
            that class is the user of this fixture (the calling
            context).

    Example:
    --------
        # This fixture should be used whenever a new answer class is
        # defined
        >>> @pytest.mark.usefixtures('answer_file')
        >>> class TestNewFrontend:
        >>>     def test1(self):
                    ...
    """
    if request.cls.__name__ in answer_files:
        answer_file = answer_files[request.cls.__name__]
        # Make sure we're not overwriting an existing answer set
        if os.path.isfile(os.path.join(answer_dir, answer_file)):
            if request.config.getoption("--answer-store"):
                raise FileExistsError(
                    "Error, attempting to overwrite "
                    "answer file {}. Either specify a new version or "
                    "set the `--force-override-answers` option".format(answer_file)
                )
    else:
        assert False
    request.cls.answer_file = answer_file


def _param_list(request):
    r"""
    Saves the non-ds, non-fixture function arguments for saving to
    the answer file.
    """
    test_params = {}
    func = request.node.function
    # co_varnames is all of the variable names local to the function
    # starting with self, then the passed args, then the vars defined
    # in the function body. This excludes fixture names
    args = func.__code__.co_varnames[1 : func.__code__.co_argcount]
    # funcargs includes the names and values of all arguments, including
    # fixtures, so we use args to weed out the fixtures. Need to have
    # special treatment of the data files loaded in fixtures for the
    # frontends
    for key, val in request.node.funcargs.items():
        if key in args and not key.startswith("ds_"):
            test_params[key] = val
    # Convert python-specific data objects (such as tuples) to a more
    # io-friendly format (in order to not have python-specific anchors
    # in the answer yaml file)
    test_params = utils._streamline_for_io(test_params)
    return test_params


@pytest.fixture(scope="function")
def hashing(request):
    r"""
    Handles initialization, generation, and saving of answer test
    result hashes.

    Answer tests that require generated data to be saved to disk
    have that data hashed. This fixture creates an empty dictionary
    as an attribute of the answer class of which the current answer
    test is a method.

    Once the test has been run and the raw data has been saved to this
    hashes dictionary, this fixture hashes the raw data and prepares
    an entry to the answer file containing the test name as well as the
    test parameter names and values to accompany the hash(es).

    These entries are then either compared to an existing entry or
    saved in the new answer file.

    Parameters:
    -----------
        request : pytest.FixtureRequest
            Provides access to the requesting test context. For
            example, if an answer class uses this fixture, such as
            TestEnzo, then request provides access to all of the
            methods and attributes of the TestEnzo class, since
            that class is the user of this fixture (the calling
            context).

    Example:
    --------
        # If every method of an answer class saves data then the
        # fixture can be applied to each method like so:
        >>> @pytest.mark.usefixtures('hashing')
        >>> class TestNewFrontend:
        >>>     def test1(self):
        >>>         ...
        >>>     def test2(self):
        >>>         ...
        # If only certain methods save data, then it must be applied
        # directly to those methods, like so:
        >>> class TestNewFrontend:
        >>>     @pytest.mark.usefixtures('hashing')
        >>>     def test1(self):
        >>>         ...
        >>>     def test2(self):
        >>>         ...
    """
    # Set up hashes dictionary
    if request.cls is not None:
        request.cls.hashes = {}
    else:
        assert False
    # Yield to the caller in order to actually perform the tests
    yield
    # Get param list
    params = _param_list(request)
    # Hash the test results
    hashes = utils._hash_results(request.cls.hashes)
    # Add the other test parameters
    hashes.update(params)
    # Add the function name as the "master" key to the hashes dict
    hashes = {request.node.name: hashes}
    # Either save or compare
    utils._handle_hashes(
        answer_dir,
        request.cls.answer_file,
        hashes,
        request.config.getoption("--answer-store"),
    )
    if request.config.getoption("--save-answer-arrays"):
        # answer_file has .yaml appended to it, but here we're saving
        # the arrays as .npy files, so we remove the .yaml extension
        utils._save_arrays(
            answer_dir,
            request.cls.answer_file.split(".")[0],
            request.cls.hashes,
            request.config.getoption("--answer-store"),
        )