def test_write_unicode_to_ascii():
def get_fake_encoding():
return 'ascii'
original_getpreferredencoding = locale.getpreferredencoding
locale.getpreferredencoding = get_fake_encoding
try:
buff = io.BytesIO()
console.color_print("μs", file=buff)
assert buff.getvalue() == b'us\n'
finally:
locale.getpreferredencoding = original_getpreferredencoding
def test_write_unicode_to_ascii():
def get_fake_encoding():
return 'ascii'
original_getpreferredencoding = locale.getpreferredencoding
locale.getpreferredencoding = get_fake_encoding
try:
buff = io.BytesIO()
console.color_print("μs", file=buff)
assert buff.getvalue() == b'us\n'
finally:
locale.getpreferredencoding = original_getpreferredencoding
def test_color_print_nofail(capfd):
# Try out color print
color_print("hello", "red")
color_print("indeed難", "blue")
with pytest.raises(ValueError):
color_print(b"really\xfe", "green", "not really")
out, err = capfd.readouterr()
assert 'hello' in out
assert 'indeed' in out
def test_color_print_nofail(capfd):
# Try out color print
color_print("hello", "red")
color_print("indeed難", "blue")
color_print(b"really\xfe", "green", "not really")
out, err = capfd.readouterr()
assert 'hello' in out
assert 'indeed' in out
assert 'really' in out
assert 'not really' in out
