84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from genesis.data_models import SheetType
|
|
from genesis.parsers.excel_parser import ExcelParseResult, ExcelParser
|
|
|
|
from tests.excel_helpers import new_workbook, save_workbook
|
|
|
|
|
|
def test_parse_table_sheets(tmp_path):
|
|
wb = new_workbook({
|
|
"機能一覧": [["機能ID", "機能名"], ["A001", "社員登録"]],
|
|
"バッチ一覧": [["バッチID", "処理名"], ["B1", "夜間集計"]],
|
|
})
|
|
path = save_workbook(tmp_path, wb)
|
|
result = ExcelParser().parse(path)
|
|
assert isinstance(result, ExcelParseResult)
|
|
by_name = {t.name: t for t in result.tables}
|
|
assert by_name["機能一覧"].detected_type == SheetType.FUNCTION
|
|
assert len(by_name["機能一覧"].rows) == 1
|
|
assert by_name["バッチ一覧"].detected_type == SheetType.BATCH
|
|
|
|
|
|
def test_parse_free_text_sheet(tmp_path):
|
|
wb = new_workbook({"メモ": [["新入社員を登録"], [], [], [], []]})
|
|
path = save_workbook(tmp_path, wb)
|
|
result = ExcelParser().parse(path)
|
|
assert len(result.tables) == 1
|
|
assert result.tables[0].extraction_method == "llm_from_free_text"
|
|
|
|
|
|
def test_parse_uses_detected_header_row(tmp_path):
|
|
# 标题行（单格）在首行，真实表头在第二行
|
|
wb = new_workbook({
|
|
"機能一覧": [["機能一覧"], ["機能ID", "機能名"], ["A001", "社員登録"]],
|
|
})
|
|
path = save_workbook(tmp_path, wb)
|
|
result = ExcelParser().parse(path)
|
|
t = result.tables[0]
|
|
assert t.headers == ["機能ID", "機能名"]
|
|
assert len(t.rows) == 1
|
|
assert t.rows[0]["機能ID"].value == "A001"
|
|
|
|
|
|
def test_parse_attaches_strikethrough_formatting(tmp_path):
|
|
from copy import copy
|
|
from openpyxl import Workbook
|
|
wb = Workbook()
|
|
ws = wb.active
|
|
ws.title = "機能一覧"
|
|
ws["A1"] = "機能ID"
|
|
ws["B1"] = "機能名"
|
|
ws["A2"] = "A001"
|
|
ws["B2"] = "社員登録"
|
|
font = copy(ws["A2"].font)
|
|
font.strike = True
|
|
ws["A2"].font = font
|
|
path = save_workbook(tmp_path, wb)
|
|
result = ExcelParser().parse(path)
|
|
cv = result.tables[0].rows[0]["機能ID"]
|
|
assert cv.formatting is not None
|
|
assert cv.formatting.strikethrough is True
|
|
|
|
|
|
from genesis.data_models import MixedParagraph, MixedSheet
|
|
from genesis.parsers.excel_parser import ExcelParseResult
|
|
|
|
|
|
def test_excel_parse_result_has_mixed_default():
|
|
r = ExcelParseResult(file_name="f.xlsx")
|
|
assert r.mixed == []
|
|
|
|
|
|
def test_mixed_paragraph_defaults():
|
|
p = MixedParagraph(kind="table")
|
|
assert p.table is None
|
|
assert p.text is None
|
|
assert p.source_range is None
|
|
|
|
|
|
def test_mixed_sheet_holds_paragraphs():
|
|
p1 = MixedParagraph(kind="table")
|
|
p2 = MixedParagraph(kind="free_text", text="备注")
|
|
ms = MixedSheet(name="混合", paragraphs=[p1, p2])
|
|
assert ms.name == "混合"
|
|
assert [p.kind for p in ms.paragraphs] == ["table", "free_text"]
|