tests/conftest.py

"""Shared fixtures and automatic test marker assignment for Brain-5D tests."""

from __future__ import annotations

from typing import TypedDict

import pytest


# Lokale Typdefinition für die Testkonfiguration
class SimulationConfig(TypedDict, total=False):
    dt_ms: float
    ticks: int
    max_delay: int
    debug_invariants: bool


class NeuronConfig(TypedDict, total=False):
    a: float
    b: float
    c: float
    d: float


class NetworkConfig(TypedDict, total=False):
    initial_connections_per_neuron: int
    neighbour_radius: float
    weight_min: float
    weight_max: float


class EnergyConfig(TypedDict, total=False):
    initial: float
    spike_cost: float
    affects_firing: bool


class TopologyConfig(TypedDict, total=False):
    input: dict[str, str | int]
    output: dict[str, str | int]
    allow_self_connections: bool
    allow_parallel_connections: bool


class DiagnosticsConfig(TypedDict, total=False):
    mode: str
    start_tick: int
    duration_ticks: int
    amplitude: float
    target_coord: list[int]
    input_plane_dim: str
    poisson_rate_hz: float
    poisson_amplitude: float


class TelemetryConfig(TypedDict, total=False):
    enabled: bool
    history_ticks: int
    spike_history_ticks: int


class VisualizationConfig(TypedDict, total=False):
    enabled: bool
    refresh_interval_ticks: int
    spike_raster_neurons: int
    projection_4d: str
    activity_tau_ticks: float


class LoggingConfig(TypedDict, total=False):
    interval_ticks: int


class Config(TypedDict, total=False):
    seed: int
    dimensions: list[int]
    initial_neurons: int
    max_neurons: int
    simulation: SimulationConfig
    neuron: NeuronConfig
    network: NetworkConfig
    energy: EnergyConfig
    topology: TopologyConfig
    diagnostics: DiagnosticsConfig
    telemetry: TelemetryConfig
    visualization: VisualizationConfig
    logging: LoggingConfig


def base_config() -> Config:
    return {
        "seed": 42,
        "dimensions": [5, 5, 5, 5, 5],
        "initial_neurons": 3,
        "max_neurons": 3125,
        "simulation": {
            "dt_ms": 1.0,
            "ticks": 20,
            "max_delay": 10,
            "debug_invariants": True,
        },
        "neuron": {"a": 0.02, "b": 0.2, "c": -65.0, "d": 8.0},
        "network": {
            "initial_connections_per_neuron": 2,
            "neighbour_radius": 2.0,
            "weight_min": 0.0,
            "weight_max": 0.5,
        },
        "energy": {"initial": 1.0, "spike_cost": 0.001, "affects_firing": False},
        "topology": {
            "input": {"dimension": "x", "coordinate": 0},
            "output": {"dimension": "x", "coordinate": 4},
            "allow_self_connections": False,
            "allow_parallel_connections": False,
        },
        "diagnostics": {
            "mode": "single_pulse",
            "start_tick": 0,
            "duration_ticks": 1,
            "amplitude": 100.0,
            "target_coord": [1, 1, 1, 1, 1],
            "input_plane_dim": "x",
            "poisson_rate_hz": 0.5,
            "poisson_amplitude": 5.0,
        },
        "telemetry": {
            "enabled": True,
            "history_ticks": 100,
            "spike_history_ticks": 100,
        },
        "visualization": {
            "enabled": False,
            "refresh_interval_ticks": 20,
            "spike_raster_neurons": 250,
            "projection_4d": "d4",
            "activity_tau_ticks": 50.0,
        },
        "logging": {"interval_ticks": 100},
    }


# ============================================================================
# Automatic test marker assignment
# ============================================================================

_CATEGORY_MAP: dict[str, str] = {
    "test_artifacts": "core",
    "test_async_storage": "storage",
    "test_auto_approval": "plasticity",
    "test_b5d_storage": "storage",
    "test_brain5d_launcher": "core",
    "test_checkpoint": "storage",
    "test_compaction": "storage",
    "test_crc": "storage",
    "test_dashboard": "dashboard",
    "test_dashboard_alpha6": "dashboard",
    "test_dashboard_alpha7": "dashboard",
    "test_dashboard_compatibility_v050a2": "dashboard",
    "test_dashboard_control_service": "dashboard",
    "test_dashboard_homeostasis": "dashboard",
    "test_dashboard_single_instance": "dashboard",
    "test_delta_journal": "storage",
    "test_eligibility": "learning",
    "test_embodiment": "embodiment",
    "test_golden_chain": "integration",
    "test_heatmap": "dashboard",
    "test_homeostasis_engine": "homeostasis",
    "test_knowledge_contracts": "core",
    "test_language_organ_contracts": "core",
    "test_lazy_storage_view": "storage",
    "test_learning_experiment": "learning",
    "test_manipulator": "core",
    "test_network": "core",
    "test_network_hooks": "core",
    "test_neuron": "core",
    "test_observatory_data": "dashboard",
    "test_optical_codec": "storage",
    "test_recovery": "storage",
    "test_research_dashboard_routes": "dashboard",
    "test_restore_continue": "storage",
    "test_reward": "learning",
    "test_runtime_control": "core",
    "test_runtime_controller_alpha4": "core",
    "test_self_organization": "plasticity",
    "test_self_organization_alpha4": "plasticity",
    "test_self_organization_policy_alpha3": "plasticity",
    "test_signal_processing_contracts": "core",
    "test_spatial_index": "core",
    "test_stdp_integration": "plasticity",
    "test_stdp_isolated": "plasticity",
    "test_storage_runtime": "storage",
    "test_structural_coordinator_alpha5": "plasticity",
    "test_structural_dashboard_routes": "dashboard",
    "test_structural_heatmap": "dashboard",
    "test_structural_journal": "storage",
    "test_structural_operator_bridge": "dashboard",
    "test_structural_plasticity_journal": "plasticity",
    "test_structural_recovery": "storage",
    "test_structural_undo": "plasticity",
    "test_research_registry": "core",
    "test_version_consistency": "core",
    "test_experiment_validity": "core",
}

_SLOW: set[str] = {
    "test_b5d_storage",
    "test_delta_journal",
    "test_golden_chain",
    "test_restore_continue",
    "test_learning_experiment",
}


def pytest_collection_modifyitems(
    config: pytest.Config, items: list[pytest.Item]
) -> None:
    """Automatically assign markers to tests based on module filename."""
    for item in items:
        node_id = item.nodeid
        module_name = node_id.split("::")[0].split("/")[-1].replace(".py", "")

        # Category marker
        cat = _CATEGORY_MAP.get(module_name)
        if cat:
            item.add_marker(getattr(pytest.mark, cat))

        # Slow marker
        if module_name in _SLOW:
            item.add_marker(pytest.mark.slow)