conftest.py

import sys
import importlib

import pytest


def pytest_collection_modifyitems(items):
    for item in items:
        item.add_marker("forked")


@pytest.fixture(autouse=True)
def setup_test():
    import veros
    from veros.logs import setup_logging

    setup_logging(loglevel="warning")
    object.__setattr__(veros.runtime_settings, "pyom_compatibility_mode", True)

    # reload all core modules to make sure changes take effect
    for name, mod in list(sys.modules.items()):
        if name.startswith("veros.core"):
            importlib.reload(mod)

    try:
        yield
    finally:
        object.__setattr__(veros.runtime_settings, "pyom_compatibility_mode", False)

    for name, mod in list(sys.modules.items()):
        if name.startswith("veros.core"):
            importlib.reload(mod)