tests/test_dump_excel.py

import pandas as pd

from utils.dump_excel import detect_dump_header_row


def test_detect_dump_header_row_prefers_second_row_when_first_is_info(monkeypatch):
    preview = pd.DataFrame(
        [
            ["Info", None, None, None],
            ["MRBTS", "LNBTS", "A3DPR", "Dist_Name"],
        ]
    )

    monkeypatch.setattr("utils.dump_excel.pd.read_excel", lambda *args, **kwargs: preview)

    header_row = detect_dump_header_row(
        "dummy.xlsb",
        "A3DPR",
        expected_columns=["MRBTS", "LNBTS", "A3DPR", "Dist_Name"],
    )

    assert header_row == 1


def test_detect_dump_header_row_keeps_first_row_when_headers_start_immediately(monkeypatch):
    preview = pd.DataFrame(
        [
            ["MRBTS", "LNBTS", "A2DPR", "id"],
            [22040, 2040, 0, 8039110],
        ]
    )

    monkeypatch.setattr("utils.dump_excel.pd.read_excel", lambda *args, **kwargs: preview)

    header_row = detect_dump_header_row(
        "dummy.xlsb",
        "A2DPR",
        expected_columns=["MRBTS", "LNBTS", "A2DPR", "id"],
    )

    assert header_row == 0