File size: 1,083 Bytes
7efc151 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | 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
|