fix: 改进

tests/kernel/algorithms/test_sm2.py

@@ -0,0 +1,135 @@
+import unittest
+from unittest.mock import patch, MagicMock
+
+from heurams.kernel.algorithms.sm2 import SM2Algorithm
+
+
+class TestSM2Algorithm(unittest.TestCase):
+    """测试 SM2Algorithm 类"""
+
+    def setUp(self):
+        # 模拟 timer 函数
+        self.timestamp_patcher = patch('heurams.kernel.algorithms.sm2.timer.get_timestamp')
+        self.daystamp_patcher = patch('heurams.kernel.algorithms.sm2.timer.get_daystamp')
+        self.mock_get_timestamp = self.timestamp_patcher.start()
+        self.mock_get_daystamp = self.daystamp_patcher.start()
+        
+        # 设置固定返回值
+        self.mock_get_timestamp.return_value = 1000.0
+        self.mock_get_daystamp.return_value = 100
+        
+    def tearDown(self):
+        self.timestamp_patcher.stop()
+        self.daystamp_patcher.stop()
+
+    def test_defaults(self):
+        """测试默认值"""
+        defaults = SM2Algorithm.defaults
+        self.assertEqual(defaults["efactor"], 2.5)
+        self.assertEqual(defaults["real_rept"], 0)
+        self.assertEqual(defaults["rept"], 0)
+        self.assertEqual(defaults["interval"], 0)
+        self.assertEqual(defaults["last_date"], 0)
+        self.assertEqual(defaults["next_date"], 0)
+        self.assertEqual(defaults["is_activated"], 0)
+        # last_modify 是动态的，仅检查存在性
+        self.assertIn("last_modify", defaults)
+
+    def test_revisor_feedback_minus_one(self):
+        """测试 feedback = -1 时跳过更新"""
+        algodata = {SM2Algorithm.algo_name: SM2Algorithm.defaults.copy()}
+        SM2Algorithm.revisor(algodata, feedback=-1)
+        # 数据应保持不变
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["efactor"], 2.5)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 0)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 0)
+
+    def test_revisor_feedback_less_than_3(self):
+        """测试 feedback < 3 重置 rept 和 interval"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 2.5, "rept": 5, "interval": 10, "real_rept": 3}}
+        SM2Algorithm.revisor(algodata, feedback=2)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 0)
+        # rept=0 导致 interval 被设置为 1
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 1)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["real_rept"], 4)  # 递增
+
+    def test_revisor_feedback_greater_equal_3(self):
+        """测试 feedback >= 3 递增 rept"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 2.5, "rept": 2, "interval": 6, "real_rept": 2}}
+        SM2Algorithm.revisor(algodata, feedback=4)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 3)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["real_rept"], 3)
+        # interval 应根据 rept 和 efactor 重新计算
+        # rept=3, interval = round(6 * 2.5) = 15
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 15)
+
+    def test_revisor_new_activation(self):
+        """测试 is_new_activation 重置 rept 和 efactor"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 3.0, "rept": 5, "interval": 20, "real_rept": 5}}
+        SM2Algorithm.revisor(algodata, feedback=5, is_new_activation=True)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["rept"], 0)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["efactor"], 2.5)
+        # interval 应为 1（因为 rept=0）
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 1)
+
+    def test_revisor_efactor_calculation(self):
+        """测试 efactor 计算"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 2.5, "rept": 1, "interval": 6, "real_rept": 1}}
+        SM2Algorithm.revisor(algodata, feedback=5)
+        # efactor = 2.5 + (0.1 - (5-5)*(0.08 + (5-5)*0.02)) = 2.5 + 0.1 = 2.6
+        self.assertAlmostEqual(algodata[SM2Algorithm.algo_name]["efactor"], 2.6, places=6)
+        
+        # 测试 efactor 下限为 1.3
+        algodata[SM2Algorithm.algo_name]["efactor"] = 1.2
+        SM2Algorithm.revisor(algodata, feedback=5)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["efactor"], 1.3)
+
+    def test_revisor_interval_calculation(self):
+        """测试 interval 计算规则"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 2.5, "rept": 0, "interval": 0, "real_rept": 0}}
+        SM2Algorithm.revisor(algodata, feedback=4)
+        # rept 从 0 递增到 1，因此 interval 应为 6
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 6)
+        
+        # 现在 rept=1，再次调用 revisor 递增到 2
+        SM2Algorithm.revisor(algodata, feedback=4)
+        # rept=2，interval = round(6 * 2.5) = 15
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["interval"], 15)
+        
+        # 单独测试 rept=1 的情况
+        algodata2 = {SM2Algorithm.algo_name: {"efactor": 2.5, "rept": 1, "interval": 0, "real_rept": 0}}
+        SM2Algorithm.revisor(algodata2, feedback=4)
+        # rept 递增到 2，interval = round(0 * 2.5) = 0
+        self.assertEqual(algodata2[SM2Algorithm.algo_name]["interval"], 0)
+
+    def test_revisor_updates_dates(self):
+        """测试更新日期字段"""
+        algodata = {SM2Algorithm.algo_name: SM2Algorithm.defaults.copy()}
+        self.mock_get_daystamp.return_value = 200
+        SM2Algorithm.revisor(algodata, feedback=5)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["last_date"], 200)
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["next_date"], 200 + algodata[SM2Algorithm.algo_name]["interval"])
+        self.assertEqual(algodata[SM2Algorithm.algo_name]["last_modify"], 1000.0)
+
+    def test_is_due(self):
+        """测试 is_due 方法"""
+        algodata = {SM2Algorithm.algo_name: {"next_date": 100}}
+        self.mock_get_daystamp.return_value = 150
+        self.assertTrue(SM2Algorithm.is_due(algodata))
+        
+        algodata[SM2Algorithm.algo_name]["next_date"] = 200
+        self.assertFalse(SM2Algorithm.is_due(algodata))
+
+    def test_rate(self):
+        """测试 rate 方法"""
+        algodata = {SM2Algorithm.algo_name: {"efactor": 2.7}}
+        self.assertEqual(SM2Algorithm.rate(algodata), "2.7")
+
+    def test_nextdate(self):
+        """测试 nextdate 方法"""
+        algodata = {SM2Algorithm.algo_name: {"next_date": 12345}}
+        self.assertEqual(SM2Algorithm.nextdate(algodata), 12345)
+
+
+if __name__ == '__main__':
+    unittest.main()