import numpy as np
from backend.data_types import LineData
from tests.node_tests._shared import make_field


def test_cross_section():
    from backend.nodes.cross_section import CrossSection
    node = CrossSection()

    N = 100
    y, x = np.mgrid[0:N, 0:N] / N
    data = x * 10.0
    field = make_field(data=data, xreal=1e-6, yreal=1e-6)

    profile, marker_pair = node.process(field, x1=0.0, y1=0.5, x2=1.0, y2=0.5, extend="none", n_samples=100)
    assert isinstance(marker_pair, tuple) and len(marker_pair) == 2
    assert isinstance(profile, LineData)
    assert len(profile) == 100
    assert profile.x_unit == field.si_unit_xy
    assert profile.y_unit == field.si_unit_z
    assert np.isclose(profile.x_axis[0], 0.0)
    assert np.isclose(profile.x_axis[-1], field.xreal)
    assert profile[0] < 0.5
    assert profile[-1] > 9.5

    profile_auto, _ = node.process(field, x1=0.0, y1=0.5, x2=1.0, y2=0.5, extend="none", n_samples=0)
    assert len(profile_auto) >= 2

    profile_ext, _ = node.process(field, x1=0.3, y1=0.5, x2=0.7, y2=0.5, extend="to_edges", n_samples=100)
    assert profile_ext[0] < 0.5
    assert profile_ext[-1] > 9.5

    profile_diag, _ = node.process(field, x1=0.0, y1=0.0, x2=1.0, y2=1.0, extend="none", n_samples=50)
    assert len(profile_diag) == 50

    from backend.nodes.cursors import Cursors
    from backend.nodes.stats import Stats

    cursors = Cursors()
    table, _ = cursors.process(profile, x1=0.25, y1=0.5, x2=0.75, y2=0.5)
    rows = {row["quantity"]: row for row in table}
    assert rows["dx"]["unit"] == field.si_unit_xy
    assert rows["dy"]["unit"] == field.si_unit_z

    from backend.execution_context import execution_callbacks, active_node
    captured = []
    with execution_callbacks(value=lambda nid, payload: captured.append(payload)), active_node("test"):
        stats = Stats()
        mean_value, = stats.process(profile, operation="mean", column="value")
        assert mean_value > 0
        assert captured[-1]["unit"] == field.si_unit_z