add assert to detect whitespace

2024-02-28 14:13:15 +08:00 · 2024-02-28 14:13:15 +08:00 · 3002cbd265
parent 95db92d601
commit 3002cbd265
1 changed files with 2 additions and 0 deletions
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@ -9,6 +9,8 @@ prompts = ['你好', "what's your name"]
 def assert_model(response):
    assert len(response) != 0
    assert 'UNUSED_TOKEN' not in response
+    assert 'Mynameis' not in response
+    assert 'Iama' not in response


 class TestChat: