Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_printer_log_friendly():
text = "This is a test."
ENV_LOG_FRIENDLY = "WASABI_LOG_FRIENDLY"
os.environ[ENV_LOG_FRIENDLY] = "True"
p = Printer(no_print=True)
if supports_ansi():
assert p.good(text) == "\u2714 This is a test."
else:
assert p.good(text) == text
del os.environ[ENV_LOG_FRIENDLY]
def test_printer_divider():
p = Printer(line_max=20, no_print=True)
p.divider() == "\x1b[1m\n================\x1b[0m"
p.divider("test") == "\x1b[1m\n====== test ======\x1b[0m"
p.divider("test", char="*") == "\x1b[1m\n****** test ******\x1b[0m"
assert (
p.divider("This is a very long text, it is very long")
== "\x1b[1m\n This is a very long text, it is very long \x1b[0m"
)
with pytest.raises(ValueError):
p.divider("test", char="~.")
def test_printer_none_encoding(monkeypatch):
"""Test that printer works even if sys.stdout.encoding is set to None. This
previously caused a very confusing error."""
monkeypatch.setattr("sys.stdout.encoding", None)
p = Printer()
def test_printer_counts():
p = Printer()
text = "This is a test."
for i in range(2):
p.good(text)
for i in range(1):
p.fail(text)
for i in range(4):
p.warn(text)
assert p.counts[MESSAGES.GOOD] == 2
assert p.counts[MESSAGES.FAIL] == 1
assert p.counts[MESSAGES.WARN] == 4
def test_printer():
p = Printer(no_print=True)
text = "This is a test."
if supports_ansi():
assert p.good(text) == "\x1b[38;5;2m\u2714 This is a test.\x1b[0m"
assert p.fail(text) == "\x1b[38;5;1m\u2718 This is a test.\x1b[0m"
assert p.warn(text) == "\x1b[38;5;3m\u26a0 This is a test.\x1b[0m"
assert p.info(text) == "\x1b[38;5;4m\u2139 This is a test.\x1b[0m"
assert p.text(text) == text
else:
assert p.good(text) == text
assert p.fail(text) == text
assert p.warn(text) == text
assert p.info(text) == text
assert p.text(text) == text
def loading_with_exception():
p = Printer()
print("\n")
with p.loading():
raise Exception("This is an error.")
def test_printer_log_friendly_prefix():
text = "This is a test."
ENV_LOG_FRIENDLY = "CUSTOM_LOG_FRIENDLY"
os.environ[ENV_LOG_FRIENDLY] = "True"
p = Printer(no_print=True, env_prefix="CUSTOM")
if supports_ansi():
assert p.good(text) == "\u2714 This is a test."
else:
assert p.good(text) == text
del os.environ[ENV_LOG_FRIENDLY]
def test_printer_loading(hide_animation):
p = Printer(hide_animation=hide_animation)
print("\n")
with p.loading("Loading..."):
time.sleep(1)
p.good("Success!")
with p.loading("Something else..."):
time.sleep(2)
p.good("Yo!")
with p.loading("Loading..."):
time.sleep(1)
p.good("Success!")
def test_printer_no_pretty():
p = Printer(no_print=True, pretty=False)
text = "This is a test."
assert p.good(text) == text
assert p.fail(text) == text
assert p.warn(text) == text
assert p.info(text) == text
assert p.text(text) == text
# coding: utf8
from __future__ import unicode_literals
from .printer import Printer # noqa
from .tables import table, row # noqa
from .traceback_printer import TracebackPrinter # noqa
from .util import color, wrap, get_raw_input, format_repr, diff_strings # noqa
from .util import MESSAGES # noqa
msg = Printer()