mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 15:28:28 +01:00
Saugat Pachhai
dba0aaeab7
I found that it's harder to style a header with the list using index, they are to be applied by header name. Otherwise, we might end up using a different header for coloring, or worse throw some error because the header name and style don't match. Row does have it's property of index, so it makes sense there.
166 lines
4.5 KiB
Python
166 lines
4.5 KiB
Python
import textwrap
|
|
|
|
import pytest
|
|
from pytest import CaptureFixture
|
|
from pytest_mock import MockerFixture
|
|
from rich.style import Style
|
|
|
|
from dvc.ui import ui
|
|
|
|
|
|
def test_plain(capsys: CaptureFixture[str]):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
)
|
|
captured = capsys.readouterr()
|
|
assert captured.out == textwrap.dedent(
|
|
"""\
|
|
first second
|
|
foo bar
|
|
foo1 bar1
|
|
foo2 bar2
|
|
"""
|
|
)
|
|
|
|
|
|
def test_plain_md(capsys: CaptureFixture[str]):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
markdown=True,
|
|
)
|
|
captured = capsys.readouterr()
|
|
assert captured.out == textwrap.dedent(
|
|
"""\
|
|
| first | second |
|
|
|---------|----------|
|
|
| foo | bar |
|
|
| foo1 | bar1 |
|
|
| foo2 | bar2 |\n
|
|
"""
|
|
)
|
|
|
|
|
|
def test_plain_pager(mocker: MockerFixture):
|
|
pager_mock = mocker.patch("dvc.utils.pager.pager")
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
pager=True,
|
|
)
|
|
|
|
pager_mock.assert_called_once_with(
|
|
textwrap.dedent(
|
|
"""\
|
|
first second
|
|
foo bar
|
|
foo1 bar1
|
|
foo2 bar2"""
|
|
)
|
|
)
|
|
|
|
|
|
def test_plain_headerless(capsys: CaptureFixture[str]):
|
|
ui.table([("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],)
|
|
captured = capsys.readouterr()
|
|
assert captured.out == textwrap.dedent(
|
|
"""\
|
|
foo bar
|
|
foo1 bar1
|
|
foo2 bar2
|
|
"""
|
|
)
|
|
|
|
|
|
def test_rich_simple(capsys: CaptureFixture[str]):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
rich_table=True,
|
|
)
|
|
# not able to test the actual style for now
|
|
captured = capsys.readouterr()
|
|
assert [
|
|
row.strip() for row in captured.out.splitlines() if row.strip()
|
|
] == ["first second", "foo bar", "foo1 bar1", "foo2 bar2"]
|
|
|
|
|
|
def test_rich_headerless(capsys: CaptureFixture[str]):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")], rich_table=True,
|
|
)
|
|
captured = capsys.readouterr()
|
|
assert [
|
|
row.strip() for row in captured.out.splitlines() if row.strip()
|
|
] == ["foo bar", "foo1 bar1", "foo2 bar2"]
|
|
|
|
|
|
def test_rich_border(capsys: CaptureFixture[str]):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
rich_table=True,
|
|
borders="simple",
|
|
)
|
|
captured = capsys.readouterr()
|
|
assert [
|
|
row.strip() for row in captured.out.splitlines() if row.strip()
|
|
] == [
|
|
"first second",
|
|
"────────────────",
|
|
"foo bar",
|
|
"foo1 bar1",
|
|
"foo2 bar2",
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"extra_opts",
|
|
[
|
|
{"header_styles": [{"style": Style(bold=True)}]},
|
|
{"header_styles": {"first": {"style": Style(bold=True)}}},
|
|
{"row_styles": [{"style": Style(bold=True)}]},
|
|
],
|
|
)
|
|
def test_rich_styles(capsys: CaptureFixture[str], extra_opts):
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
rich_table=True,
|
|
**extra_opts
|
|
)
|
|
# not able to test the actual style for now
|
|
captured = capsys.readouterr()
|
|
assert [
|
|
row.strip() for row in captured.out.splitlines() if row.strip()
|
|
] == ["first second", "foo bar", "foo1 bar1", "foo2 bar2"]
|
|
|
|
|
|
def test_rich_pager(mocker: MockerFixture):
|
|
pager_mock = mocker.patch("dvc.utils.pager.pager")
|
|
|
|
ui.table(
|
|
[("foo", "bar"), ("foo1", "bar1"), ("foo2", "bar2")],
|
|
headers=["first", "second"],
|
|
rich_table=True,
|
|
pager=True,
|
|
)
|
|
received_text = pager_mock.call_args[0][0]
|
|
assert [
|
|
row.strip() for row in received_text.splitlines() if row.strip()
|
|
] == ["first second", "foo bar", "foo1 bar1", "foo2 bar2"]
|
|
|
|
|
|
@pytest.mark.parametrize("rich_table", [True, False])
|
|
def test_empty(capsys: CaptureFixture[str], rich_table: str):
|
|
ui.table([], rich_table=rich_table)
|
|
out, err = capsys.readouterr()
|
|
assert (out, err) == ("", "")
|
|
|
|
|
|
def test_empty_markdown(capsys: CaptureFixture[str]):
|
|
ui.table([], headers=["Col1", "Col2"], markdown=True)
|
|
out, err = capsys.readouterr()
|
|
assert (out, err) == ("| Col1 | Col2 |\n|--------|--------|\n\n", "")
|