test_merged_cells.py

import os

from pyexcel_xlsx import get_data
from pyexcel_xlsx.xlsxr import MergedCell
from openpyxl.worksheet.cell_range import CellRange

from nose.tools import eq_


def test_merged_cells():
    data = get_data(
        get_fixture("merged-cell-sheet.xlsx"),
        detect_merged_cells=True,
        library="pyexcel-xlsx",
    )
    expected = [[1, 2, 3], [1, 5, 6], [1, 8, 9], [10, 11, 11]]
    eq_(data["Sheet1"], expected)


def test_complex_merged_cells():
    data = get_data(
        get_fixture("complex-merged-cells-sheet.xlsx"),
        detect_merged_cells=True,
        library="pyexcel-xlsx",
    )
    expected = [
        [1, 1, 2, 3, 15, 16, 22, 22, 24, 24],
        [1, 1, 4, 5, 15, 17, 22, 22, 24, 24],
        [6, 7, 8, 9, 15, 18, 22, 22, 24, 24],
        [10, 11, 11, 12, 19, 19, 23, 23, 24, 24],
        [13, 11, 11, 14, 20, 20, 23, 23, 24, 24],
        [21, 21, 21, 21, 21, 21, 23, 23, 24, 24],
        [25, 25, 25, 25, 25, 25, 25, 25, 25, 25],
        [25, 25, 25, 25, 25, 25, 25, 25, 25, 25],
    ]
    eq_(data["Sheet1"], expected)


def test_exploration():
    data = get_data(
        get_fixture("merged-sheet-exploration.xlsx"),
        detect_merged_cells=True,
        library="pyexcel-xlsx",
    )
    expected_sheet1 = [
        [1, 1, 1, 1, 1, 1],
        [2],
        [2],
        [2],
        [2],
        [2],
        [2],
        [2],
        [2],
        [2],
    ]
    eq_(data["Sheet1"], expected_sheet1)
    expected_sheet2 = [[3], [3], [3], [3, 4, 4, 4, 4, 4, 4], [3], [3], [3]]
    eq_(data["Sheet2"], expected_sheet2)
    expected_sheet3 = [
        ["", "", "", "", "", 2, 2, 2],
        [],
        [],
        [],
        ["", "", "", 5],
        ["", "", "", 5],
        ["", "", "", 5],
        ["", "", "", 5],
        ["", "", "", 5],
    ]
    eq_(data["Sheet3"], expected_sheet3)


def test_merged_cell_class():
    test_dict = {}
    merged_cell = MergedCell(CellRange(range_string="A7:J8"))
    merged_cell.register_cells(test_dict)
    keys = sorted(list(test_dict.keys()))
    expected = [
        "7-1",
        "7-10",
        "7-2",
        "7-3",
        "7-4",
        "7-5",
        "7-6",
        "7-7",
        "7-8",
        "7-9",
        "8-1",
        "8-10",
        "8-2",
        "8-3",
        "8-4",
        "8-5",
        "8-6",
        "8-7",
        "8-8",
        "8-9",
    ]
    eq_(keys, expected)
    eq_(merged_cell, test_dict["7-1"])
    eq_(merged_cell.bottom_row(), 8)
    eq_(merged_cell.right_column(), 10)


def get_fixture(file_name):
    return os.path.join("tests", "fixtures", file_name)