feat: 进一步改进

This commit is contained in:
2025-12-31 00:57:07 +08:00
parent b65dad6a1f
commit eaa38fb880
43 changed files with 251 additions and 528 deletions

View File

@@ -174,7 +174,7 @@ class TestSM2Algorithm(unittest.TestCase):
def test_rate(self):
"""测试 rate 方法"""
algodata = {SM2Algorithm.algo_name: {"efactor": 2.7}}
self.assertEqual(SM2Algorithm.rate(algodata), "2.7")
self.assertEqual(SM2Algorithm.get_rating(algodata), "2.7")
def test_nextdate(self):
"""测试 nextdate 方法"""

View File

@@ -98,7 +98,7 @@ class TestElectron(unittest.TestCase):
electron = Electron("test_electron")
with patch.object(electron.algo, "rate") as mock_rate:
mock_rate.return_value = "good"
result = electron.get_rate()
result = electron.get_rating()
mock_rate.assert_called_once_with(electron.algodata)
self.assertEqual(result, "good")

View File

@@ -1,7 +1,7 @@
import unittest
from unittest.mock import Mock
from heurams.kernel.puzzles.base import BasePuzzle
from heurams.kernel.evaluators.base import BasePuzzle
class TestBasePuzzle(unittest.TestCase):

View File

@@ -1,7 +1,7 @@
import unittest
from unittest.mock import MagicMock, patch
from heurams.kernel.puzzles.cloze import ClozePuzzle
from heurams.kernel.evaluators.cloze import ClozePuzzle
class TestClozePuzzle(unittest.TestCase):

View File

@@ -1,7 +1,7 @@
import unittest
from unittest.mock import MagicMock, call, patch
from heurams.kernel.puzzles.mcq import MCQPuzzle
from heurams.kernel.evaluators.mcq import MCQPuzzle
class TestMCQPuzzle(unittest.TestCase):