From f7a4bea785deb6906d7ce8a95618e08c8ba2f5ec Mon Sep 17 00:00:00 2001
From: Echo-Nie <157974576+Echo-Nie@users.noreply.github.com>
Date: Wed, 3 Sep 2025 10:55:02 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90Hackathon=209th=20No.84=E3=80=91Supple?=
 =?UTF-8?q?mentary=20Unit=20Test=20for=20fastdeploy/reasoning=20(#3570)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

测试内容：测试基类的注册、获取函数功能是否正常

Co-authored-by: Tao Luo <luotao02@baidu.com>
---
 tests/reasoning/test_reasoning_parser.py | 107 +++++++++++++++++++++++
 1 file changed, 107 insertions(+)
 create mode 100644 tests/reasoning/test_reasoning_parser.py

diff --git a/tests/reasoning/test_reasoning_parser.py b/tests/reasoning/test_reasoning_parser.py
new file mode 100644
index 000000000..2ef56e308
--- /dev/null
+++ b/tests/reasoning/test_reasoning_parser.py
@@ -0,0 +1,107 @@
+"""
+# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+
+import unittest
+
+from fastdeploy.reasoning import ReasoningParser, ReasoningParserManager
+
+
+class TestReasoningParser(ReasoningParser):
+    def is_reasoning_end(self, input_ids):
+        """
+        Return True to simulate end of reasoning content.
+        """
+        return True
+
+    def extract_content_ids(self, input_ids):
+        """
+        Return input_ids directly for testing.
+        """
+        return input_ids
+
+    def extract_reasoning_content(self, model_output, request):
+        """
+        Used for testing non-streaming extraction.
+        """
+        return model_output, model_output
+
+    def extract_reasoning_content_streaming(
+        self, previous_text, current_text, delta_text, previous_token_ids, current_token_ids, delta_token_ids
+    ):
+        """
+        Return None for streaming extraction; minimal implementation for testing.
+        """
+        return None
+
+
+class TestReasoningParserManager(unittest.TestCase):
+    """
+    Unit tests for ReasoningParserManager functionality.
+    """
+
+    def setUp(self):
+        """
+        Save original registry to restore after each test.
+        """
+        self.original_parsers = ReasoningParserManager.reasoning_parsers.copy()
+
+    def tearDown(self):
+        """
+        Restore original registry to avoid test pollution.
+        """
+        ReasoningParserManager.reasoning_parsers = self.original_parsers.copy()
+
+    def test_register_and_get_parser(self):
+        """
+        Test that a parser can be registered and retrieved successfully.
+        Verifies normal registration and retrieval functionality.
+        """
+        ReasoningParserManager.register_module(module=TestReasoningParser, name="test_parser", force=True)
+        parser_cls = ReasoningParserManager.get_reasoning_parser("test_parser")
+        self.assertIs(parser_cls, TestReasoningParser)
+
+    def test_register_duplicate_without_force_raises(self):
+        """
+        Test that registering a parser with an existing name without force raises KeyError.
+        Ensures duplicate registrations are handled correctly.
+        """
+        ReasoningParserManager.register_module(module=TestReasoningParser, name="test_parser2", force=True)
+        with self.assertRaises(KeyError):
+            ReasoningParserManager.register_module(module=TestReasoningParser, name="test_parser2", force=False)
+
+    def test_register_non_subclass_raises(self):
+        """
+        Test that registering a class not inheriting from ReasoningParser raises TypeError.
+        Ensures type safety for registered modules.
+        """
+
+        class NotParser:
+            pass
+
+        with self.assertRaises(TypeError):
+            ReasoningParserManager.register_module(module=NotParser, name="not_parser")
+
+    def test_get_unregistered_parser_raises(self):
+        """
+        Test that retrieving a parser that was not registered raises KeyError.
+        Ensures get_reasoning_parser handles unknown names correctly.
+        """
+        with self.assertRaises(KeyError):
+            ReasoningParserManager.get_reasoning_parser("nonexistent_parser")
+
+
+if __name__ == "__main__":
+    unittest.main()