chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

configure utf8 encoding

Browse Source
This commit is contained in:
qingxu fu 2023-08-16 21:29:16 +08:00
parent dc7fab3c19
commit 00f63cb0bc
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_utils.py
View File

@ -22,10 +22,12 @@ def silence_stdout(func):
def wrapper(*args, **kwargs):
_original_stdout = sys.stdout
sys.stdout = open(os.devnull, 'w')
sys.stdout.reconfigure(encoding='utf-8')
for q in func(*args, **kwargs):
sys.stdout = _original_stdout
yield q
sys.stdout = open(os.devnull, 'w')
sys.stdout.reconfigure(encoding='utf-8')
sys.stdout.close()
sys.stdout = _original_stdout
return wrapper
@ -35,6 +37,7 @@ def silence_stdout_fn(func):
def wrapper(*args, **kwargs):
_original_stdout = sys.stdout
sys.stdout = open(os.devnull, 'w')
sys.stdout.reconfigure(encoding='utf-8')
result = func(*args, **kwargs)
sys.stdout.close()
sys.stdout = _original_stdout