From 3002cbd2658dec447754ee014e21134fd5dfde09 Mon Sep 17 00:00:00 2001
From: zhulin1 <zhulin1@pjlab.org.cn>
Date: Wed, 28 Feb 2024 14:13:15 +0800
Subject: [PATCH] add assert to detect whitespace

---
 tests/test_hf_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index f8d30d3..ec93217 100644
--- 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: