github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add assert to detect whitespace

pull/717/head
zhulin1 2024-02-28 14:13:15 +08:00
parent 95db92d601
commit 3002cbd265
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_hf_model.py
View File

@ -9,6 +9,8 @@ prompts = ['你好', "what's your name"]
def assert_model(response): def assert_model(response):
assert len(response) != 0 assert len(response) != 0
assert 'UNUSED_TOKEN' not in response assert 'UNUSED_TOKEN' not in response
assert 'Mynameis' not in response
assert 'Iama' not in response
class TestChat: class TestChat: