feat: 进一步改进
This commit is contained in:
@@ -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 方法"""
|
||||
|
||||
@@ -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")
|
||||
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user