Colin Maudry
e5ca7d62a3
- I-1: Add explicit lower bounds to width assertions (>= 15 for autre, >= 40 for objet) - I-2: Wrap write_excel() and close() in try/finally to ensure cleanup even on exception - M-2: Add openpyxl to dev dependencies in pyproject.toml Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
import io
|
|
|
|
import openpyxl
|
|
import polars as pl
|
|
|
|
from src.utils.table import write_styled_excel
|
|
|
|
|
|
def test_write_styled_excel_header_bold_and_color():
|
|
df = pl.DataFrame({"objet": ["marché A"], "montant": [1000.0]})
|
|
buf = io.BytesIO()
|
|
write_styled_excel(df, buf)
|
|
buf.seek(0)
|
|
wb = openpyxl.load_workbook(buf)
|
|
ws = wb.active
|
|
cell = ws.cell(row=1, column=1)
|
|
assert cell.font.bold is True
|
|
assert cell.fill.fgColor.rgb == "FFB33821"
|
|
assert cell.font.color.rgb == "FFFFFFFF"
|
|
|
|
|
|
def test_write_styled_excel_data_cell_text_wrap():
|
|
df = pl.DataFrame({"objet": ["marché A"]})
|
|
buf = io.BytesIO()
|
|
write_styled_excel(df, buf)
|
|
buf.seek(0)
|
|
wb = openpyxl.load_workbook(buf)
|
|
ws = wb.active
|
|
cell = ws.cell(row=2, column=1)
|
|
assert cell.alignment.wrap_text is True
|
|
|
|
|
|
def test_write_styled_excel_known_column_wider_than_minimum():
|
|
# "objet" a une largeur explicite (350px) > minimum (132px)
|
|
# "autre" n'a pas de largeur explicite → reçoit le minimum
|
|
df = pl.DataFrame({"autre": ["x"], "objet": ["y"]})
|
|
buf = io.BytesIO()
|
|
write_styled_excel(df, buf)
|
|
buf.seek(0)
|
|
wb = openpyxl.load_workbook(buf)
|
|
ws = wb.active
|
|
width_autre = ws.column_dimensions["A"].width # colonne 1 = "autre"
|
|
width_objet = ws.column_dimensions["B"].width # colonne 2 = "objet"
|
|
assert width_autre >= 15 # 132px minimum → ~18.9 chars xlsxwriter, borne basse 15
|
|
assert width_objet >= 40 # 350px → ~50 chars xlsxwriter, borne basse 40
|
|
|
|
|
|
def test_write_styled_excel_custom_worksheet_name():
|
|
df = pl.DataFrame({"col": ["val"]})
|
|
buf = io.BytesIO()
|
|
write_styled_excel(df, buf, worksheet="2025")
|
|
buf.seek(0)
|
|
wb = openpyxl.load_workbook(buf)
|
|
assert "2025" in wb.sheetnames
|