mirror of https://github.com/Textualize/rich.git
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import pytest
|
|
|
|
from rich.box import get_safe_box, ASCII, DOUBLE, ROUNDED, HEAVY, SQUARE
|
|
|
|
|
|
def test_str():
|
|
assert str(ASCII) == "+--+\n| ||\n|-+|\n| ||\n|-+|\n|-+|\n| ||\n+--+\n"
|
|
|
|
|
|
def test_repr():
|
|
assert repr(ASCII) == "Box(...)"
|
|
|
|
|
|
def test_get_top():
|
|
top = HEAVY.get_top(widths=[1, 2])
|
|
assert top == "┏━┳━━┓"
|
|
|
|
|
|
def test_get_row():
|
|
head_row = DOUBLE.get_row(widths=[3, 2, 1], level="head")
|
|
assert head_row == "╠═══╬══╬═╣"
|
|
|
|
row = ASCII.get_row(widths=[1, 2, 3], level="row")
|
|
assert row == "|-+--+---|"
|
|
|
|
foot_row = ROUNDED.get_row(widths=[2, 1, 3], level="foot")
|
|
assert foot_row == "├──┼─┼───┤"
|
|
|
|
with pytest.raises(ValueError):
|
|
ROUNDED.get_row(widths=[1, 2, 3], level="FOO")
|
|
|
|
|
|
def test_get_bottom():
|
|
bottom = HEAVY.get_bottom(widths=[1, 2, 3])
|
|
assert bottom == "┗━┻━━┻━━━┛"
|
|
|
|
|
|
def test_get_safe_box():
|
|
assert get_safe_box(HEAVY, True) == SQUARE
|
|
assert get_safe_box(HEAVY, False) == HEAVY
|
|
assert get_safe_box(None, True) is None
|
|
assert get_safe_box(None, False) is None
|