mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
pre-commit-ci[bot]andSaugat Pachhai
3375a7c39f
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/astral-sh/ruff-pre-commit: v0.3.3 → v0.3.4](https://github.com/astral-sh/ruff-pre-commit/compare/v0.3.3...v0.3.4) * fixes --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Saugat Pachhai (सौगात) <[email protected]>
67 lines
1.5 KiB
Python
67 lines
1.5 KiB
Python
import datetime
|
|
import textwrap
|
|
|
|
import pytest
|
|
|
|
from dvc.ui import Console
|
|
|
|
|
|
def test_write(capsys):
|
|
"""Test that ui.write works."""
|
|
console = Console(enable=True)
|
|
message = "hello world"
|
|
console.write(message)
|
|
console.error_write(message)
|
|
|
|
captured = capsys.readouterr()
|
|
assert captured.out == f"{message}\n"
|
|
assert captured.err == f"{message}\n"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"isatty, expected_output",
|
|
[
|
|
(
|
|
True,
|
|
textwrap.dedent(
|
|
"""\
|
|
{
|
|
"hello": "world",
|
|
"date": "1970-01-01 00:00:00"
|
|
}
|
|
"""
|
|
),
|
|
),
|
|
(
|
|
False,
|
|
textwrap.dedent(
|
|
"""\
|
|
{"hello": "world", "date": "1970-01-01 00:00:00"}
|
|
"""
|
|
),
|
|
),
|
|
],
|
|
)
|
|
def test_write_json(capsys, mocker, isatty, expected_output):
|
|
"""Test that ui.write_json works."""
|
|
|
|
console = Console(enable=True)
|
|
mocker.patch.object(console, "isatty", return_value=isatty)
|
|
message = {"hello": "world", "date": datetime.datetime(1970, 1, 1)} # noqa: DTZ001
|
|
console.write_json(message, default=str)
|
|
captured = capsys.readouterr()
|
|
assert captured.out == expected_output
|
|
|
|
|
|
def test_capsys_works(capsys):
|
|
"""Sanity check that capsys can capture outputs from a global ui."""
|
|
from dvc.ui import ui
|
|
|
|
message = "hello world"
|
|
ui.write(message)
|
|
ui.error_write(message)
|
|
|
|
captured = capsys.readouterr()
|
|
assert captured.out == f"{message}\n"
|
|
assert captured.err == f"{message}\n"
|