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