import pytest


def pytest_collection_modifyitems(items):
    for item in items:
        if "unit" in item.name:
            item.add_marker(pytest.mark.unit)
        if "integrate" in item.name:
            item.add_marker(pytest.mark.integrate)
