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"