import pandas as pd
import pytest


def _generate_example_data(exposures):
    values = [(i, i * 10, i * 100) for i in range(1, exposures + 1)]
    columns = ["Exposure.Id", "Bkg.Median", "Spot.Median"]

    return pd.DataFrame(values, columns=columns)


@pytest.fixture()
def classic_example_data():
    return _generate_example_data(4)


@pytest.fixture()
def regression_example_data():
    return _generate_example_data(6)


def test_add_exposure_info_classic(classic_example_data):
    from conda_helpers.calibration import add_exposure_info

    result = add_exposure_info(classic_example_data)

    assert list(result["Exposure.Channel"]) == ["Cy3"] * 2 + ["Cy5"] * 2
    assert list(result["Exposure.Time"]) == [100, 10, 60, 6]


def test_add_exposure_info_regression(regression_example_data):
    from conda_helpers.calibration import add_exposure_info

    result = add_exposure_info(regression_example_data)

    assert list(result["Exposure.Channel"]) == ["Cy3"] * 3 + ["Cy5"] * 3
    assert list(result["Exposure.Time"]) == [100, 30, 10, 60, 18, 6]


def test_remove_non_signal_positions():
    from conda_helpers.calibration import remove_non_signal_positions

    data = pd.DataFrame(range(1, 12), columns=["Pos.Id"])

    result = remove_non_signal_positions(data)

    assert set(result["Pos.Id"]) == {3, 4, 5, 6, 7, 8}


def test_add_signal_columns(classic_example_data):
    from conda_helpers.calibration import add_signal_columns

    result = add_signal_columns(classic_example_data)

    assert len(result) == 2 * len(classic_example_data)

    assert (
        list(result["Signal.Source"])
        == ["Spot.Median"] * 4 + ["Bkg.Median"] * 4
    )
    assert list(result["Signal.Intensity"]) == [
        100,
        200,
        300,
        400,
        10,
        20,
        30,
        40,
    ]


def test_prepare_calibration_data(classic_example_data):
    from conda_helpers.calibration import prepare_calibration_data

    classic_example_data["Pos.Id"] = 0
    result = prepare_calibration_data(classic_example_data)

    assert len(result) == 2 * len(classic_example_data)
    assert "Exposure.Channel" in result.columns
    assert "Exposure.Time" in result.columns
    assert "Signal.Source" in result.columns
    assert "Signal.Intensity" in result.columns