mirror of
https://github.com/Textualize/rich.git
synced 2025-07-07 21:04:58 +00:00
0177b03700
* fix: minor black formatting * style: update CONTRIBUTORS.md * fix: more black formatting Default black with no arguments provided --------- Co-authored-by: Darren Burns <darrenburns@users.noreply.github.com>
37 lines
917 B
Python
37 lines
917 B
Python
import io
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from rich.console import Console
|
|
from rich.file_proxy import FileProxy
|
|
|
|
|
|
def test_empty_bytes():
|
|
console = Console()
|
|
file_proxy = FileProxy(console, sys.stdout)
|
|
# File should raise TypeError when writing bytes
|
|
with pytest.raises(TypeError):
|
|
file_proxy.write(b"") # type: ignore
|
|
with pytest.raises(TypeError):
|
|
file_proxy.write(b"foo") # type: ignore
|
|
|
|
|
|
def test_flush():
|
|
file = io.StringIO()
|
|
console = Console(file=file)
|
|
file_proxy = FileProxy(console, file)
|
|
file_proxy.write("foo")
|
|
assert file.getvalue() == ""
|
|
file_proxy.flush()
|
|
assert file.getvalue() == "foo\n"
|
|
|
|
|
|
def test_new_lines():
|
|
file = io.StringIO()
|
|
console = Console(file=file)
|
|
file_proxy = FileProxy(console, file)
|
|
file_proxy.write("-\n-")
|
|
assert file.getvalue() == "-\n"
|
|
file_proxy.flush()
|
|
assert file.getvalue() == "-\n-\n"
|