135 lines
6.3 KiB
Python
135 lines
6.3 KiB
Python
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() |