diff --git a/test_simple.py b/test_simple.py new file mode 100644 index 0000000..21be3aa --- /dev/null +++ b/test_simple.py @@ -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() \ No newline at end of file