parent
bcf3d9fe4c
commit
c32972a11f
@ -0,0 +1,27 @@
|
||||
import unittest
|
||||
from simple_danmu_analyzer import SimpleDanmuAnalyzer
|
||||
|
||||
|
||||
class TestSimpleDanmuAnalyzer(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.analyzer = SimpleDanmuAnalyzer()
|
||||
self.analyzer.danmu_list = [
|
||||
'大语言模型很强大',
|
||||
'AI改变世界',
|
||||
'机器学习很有趣',
|
||||
'大语言模型应用',
|
||||
'深度学习技术'
|
||||
]
|
||||
|
||||
def test_analyze_words(self):
|
||||
result = self.analyzer.analyze_words(3)
|
||||
self.assertEqual(len(result), 3)
|
||||
|
||||
def test_get_conclusions(self):
|
||||
result = self.analyzer.get_conclusions()
|
||||
self.assertIn("共分析", result)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Loading…
Reference in new issue