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: