81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""tests for software_columns
|
||
|
|
|
||
|
|
:author: Benito López and the python-escpos developers
|
||
|
|
:organization: `python-escpos <https://github.com/python-escpos>`_
|
||
|
|
:copyright: Copyright (c) 2024 `python-escpos <https://github.com/python-escpos>`_
|
||
|
|
:license: MIT
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_rearrange_into_cols(driver) -> None:
|
||
|
|
"""
|
||
|
|
GIVEN a list of columnable text
|
||
|
|
WHEN the column width is different for each column and some strings exceed the max width
|
||
|
|
THEN check the strings are properly wrapped, truncated and rearranged into some columns
|
||
|
|
"""
|
||
|
|
|
||
|
|
output = driver._rearrange_into_cols(
|
||
|
|
text_list=["fits", "row1 row2", "truncate and wrap"], widths=[4, 5, 6]
|
||
|
|
)
|
||
|
|
assert output == [["fits", "row1", "trunc."], ["", "row2", "and"], ["", "", "wrap"]]
|
||
|
|
|
||
|
|
|
||
|
|
def test_add_padding_into_cols(driver) -> None:
|
||
|
|
"""
|
||
|
|
GIVEN a list of strings
|
||
|
|
WHEN adding padding and different alignments to each string
|
||
|
|
THEN check the strings are correctly padded and aligned
|
||
|
|
"""
|
||
|
|
|
||
|
|
output = driver._add_padding_into_cols(
|
||
|
|
text_list=["col1", "col2", "col3"],
|
||
|
|
widths=[6, 6, 6],
|
||
|
|
align=["center", "left", "right"],
|
||
|
|
)
|
||
|
|
assert output == [" col1 ", "col2 ", " col3"]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("text_list", ["", [], None])
|
||
|
|
@pytest.mark.parametrize("widths", [30.5, "30", None])
|
||
|
|
@pytest.mark.parametrize("align", ["invalid_align_name", "", None])
|
||
|
|
def test_software_columns_invalid_args(driver, text_list, widths, align) -> None:
|
||
|
|
"""
|
||
|
|
GIVEN a dummy printer object
|
||
|
|
WHEN non valid params are passed
|
||
|
|
THEN check raise exception
|
||
|
|
"""
|
||
|
|
bad_text_list = {"text_list": text_list, "widths": 5, "align": "left"}
|
||
|
|
bad_widths = {"text_list": ["valid"], "widths": widths, "align": "left"}
|
||
|
|
bad_align = {"text_list": ["valid"], "widths": 5, "align": align}
|
||
|
|
|
||
|
|
bad_args = [bad_text_list, bad_widths, bad_align]
|
||
|
|
for kwargs in bad_args:
|
||
|
|
with pytest.raises(Exception):
|
||
|
|
driver.software_columns(**kwargs)
|
||
|
|
driver.close()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"text_list",
|
||
|
|
[
|
||
|
|
["col1", "col2", "col3"],
|
||
|
|
["wrap this string", "wrap this string", "wrap this string"],
|
||
|
|
["truncate_this_string", "truncate_this_string", "truncate_this_string"],
|
||
|
|
],
|
||
|
|
)
|
||
|
|
@pytest.mark.parametrize("widths", [[10, 10, 10], [10], 30])
|
||
|
|
@pytest.mark.parametrize("align", [["center", "left", "right"], ["center"], "center"])
|
||
|
|
def test_software_columns_valid_args(driver, text_list, widths, align) -> None:
|
||
|
|
"""
|
||
|
|
GIVEN a dummy printer object
|
||
|
|
WHEN valid params are passed
|
||
|
|
THEN check no errors
|
||
|
|
"""
|
||
|
|
driver.software_columns(text_list=text_list, widths=widths, align=align)
|
||
|
|
driver.close()
|