"""
Module containing fixtures and hooks used by the pytest test suite.
"""

import re

import numpy as np
import pytest

from brian2.core.clocks import defaultclock
from brian2.core.functions import DEFAULT_FUNCTIONS, Function
from brian2.devices import get_device, reinit_devices
from brian2.devices.device import reinit_and_delete, set_device
from brian2.units import ms


def pytest_ignore_collect(collection_path, config):
    if config.option.doctestmodules:
        if "tests" in collection_path.parts:
            return True  # Ignore tests package for doctests
    # Do not test brian2.hears bridge (needs Brian1)
    if collection_path.name == "hears.py":
        return True


# The "random" values are always 0.5
def fake_randn(vectorisation_idx):
    return 0.5 * np.ones_like(vectorisation_idx)


fake_randn = Function(
    fake_randn, arg_units=[], return_unit=1, auto_vectorise=True, stateless=False
)
fake_randn.implementations.add_implementation(
    "cpp",
    """
                                              double randn(int vectorisation_idx)
                                              {
                                                  return 0.5;
                                              }
                                              """,
)
fake_randn.implementations.add_implementation(
    "cython",
    """
                                    cdef double randn(int vectorisation_idx):
                                        return 0.5
                                    """,
)


@pytest.fixture
def fake_randn_randn_fixture():
    orig_randn = DEFAULT_FUNCTIONS["randn"]
    DEFAULT_FUNCTIONS["randn"] = fake_randn
    yield None
    DEFAULT_FUNCTIONS["randn"] = orig_randn


# Fixture that is used for all tests
@pytest.fixture(autouse=True)
def setup_and_teardown(request):
    # Set preferences before each test
    import brian2

    if hasattr(request.config, "workerinput"):
        config = request.config.workerinput
        for key, value in config["brian_prefs"].items():
            if isinstance(value, tuple) and value[0] == "TYPE":
                matches = re.match(r"<(type|class) 'numpy\.(.+)'>", value[1])
                if matches is None or len(matches.groups()) != 2:
                    raise TypeError(
                        f"Do not know how to handle {value[1]} in preferences"
                    )
                t = matches.groups()[1]
                if t == "float64":
                    value = np.float64
                elif t == "float32":
                    value = np.float32
                elif t == "int64":
                    value = np.int64
                elif t == "int32":
                    value = np.int32

            brian2.prefs[key] = value
        set_device(config["device"], **config["device_options"])
    else:
        for k, v in request.config.brian_prefs.items():
            brian2.prefs[k] = v
        set_device(request.config.device, **request.config.device_options)
    brian2.prefs._backup()
    # Print output changed in numpy 1.14, stick with the old format to
    # avoid doctest failures
    try:
        np.set_printoptions(legacy="1.13")
    except TypeError:
        pass  # using a numpy version < 1.14

    yield  # run test

    # Reset defaultclock.dt to be sure
    defaultclock.dt = 0.1 * ms


# (Optionally) mark tests raising NotImplementedError as skipped (mostly used
# for testing Brian2GeNN)
@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item, call):
    if hasattr(item.config, "workerinput"):
        fail_for_not_implemented = item.config.workerinput["fail_for_not_implemented"]
    else:
        fail_for_not_implemented = item.config.fail_for_not_implemented
    outcome = yield
    rep = outcome.get_result()
    if rep.outcome == "failed":
        project_dir = getattr(get_device(), "project_dir", None)
        if project_dir is not None:
            rep.sections.append(("Standalone project directory", f"{project_dir}"))
        reinit_devices()
        if not fail_for_not_implemented:
            exc_cause = getattr(call.excinfo.value, "__cause__", None)
            if call.excinfo.errisinstance(NotImplementedError) or isinstance(
                exc_cause, NotImplementedError
            ):
                rep.outcome = "skipped"
                r = call.excinfo._getreprcrash()
                rep.longrepr = (str(r.path), r.lineno, r.message)
    else:
        # clean up after the test (delete directory for standalone)
        reinit_and_delete()