Ce add bad cases (#3215)

* add repitation early stop cases

* add repitation early stop cases

* add bad cases

* add bad cases

test/ce/server/test_DDoS.py

@@ -0,0 +1,8 @@
+#!/bin/env python3
+# -*- coding: utf-8 -*-
+# @author DDDivano
+# encoding=utf-8 vi:ts=4:sw=4:expandtab:ft=python
+
+"""
+分布式拒绝服务
+"""

test/ce/server/test_base_chat.py

@@ -104,6 +104,9 @@ def test_stop_sequence():
 
 
 def test_stop_sequence1():
+    """
+    不加stop看看是否有影响
+    """
     data = {
         "stream": False,
         "messages": [
@@ -122,6 +125,53 @@ def test_stop_sequence1():
     assert "第二段" in content
 
 
+def test_stop_sequence2():
+    """
+    stop token长度测试
+    """
+    data = {
+        "stream": False,
+        "stop": ["这是第二段啦啦"],
+        "messages": [
+            {
+                "role": "user",
+                "content": "你要严格按照我接下来的话输出，输出冒号后面的内容，请输出：这是第一段。果冻这是第二段啦啦啦啦啦。",
+            },
+        ],
+        "max_tokens": 50,
+        "top_p": 0,
+    }
+    payload = build_request_payload(TEMPLATE, data)
+    resp = send_request(URL, payload).json()
+    content = resp["choices"][0]["message"]["content"]
+    # token_list = get_token_list(resp)
+    print("截断输出:", content)
+    assert "啦啦啦" not in content
+
+
+# def test_stop_sequence3():
+#     """
+#     stop token 数量测试
+#     """
+#     data = {
+#         "stream": False,
+#         "stop": ["。", "果冻", "果", "冻", "第二", "二"],
+#         "messages": [
+#             {
+#                 "role": "user",
+#                 "content": "你要严格按照我接下来的话输出，输出冒号后面的内容，请输出：这是第一段。果冻这是第二段啦啦啦啦啦。",
+#             },
+#         ],
+#         "max_tokens": 50,
+#         "top_p": 0,
+#     }
+#     payload = build_request_payload(TEMPLATE, data)
+#     resp = send_request(URL, payload).json()
+#     content = resp["choices"][0]["message"]["content"]
+#     print("截断输出:", content)
+#     assert "啦啦啦" not in content
+
+
 def test_sampling_parameters():
     data = {
         "stream": False,

test/ce/server/test_evil_cases.py

@@ -0,0 +1,7 @@
+#!/bin/env python3
+# -*- coding: utf-8 -*-
+# @author DDDivano
+# encoding=utf-8 vi:ts=4:sw=4:expandtab:ft=python
+"""
+边缘检测 ，攻击性测试
+"""