You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
671 B
21 lines
671 B
import unittest
|
|
import to_excel
|
|
import os
|
|
import sys
|
|
# 获取当前文件所在目录的绝对路径
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# 获取 developed_code 目录的绝对路径，并添加到 sys.path 中
|
|
developed_code_path = os.path.abspath(os.path.join(current_dir, '..'))
|
|
sys.path.append(developed_code_path)
|
|
|
|
class TestToExcel(unittest.TestCase):
|
|
|
|
def test_filter_and_count_danmu(self):
|
|
danmu_list = ["AI 是未来", "机器学习", "AI 是未来", "人工智能"]
|
|
result = to_excel.filter_and_count_danmu(danmu_list)
|
|
self.assertEqual(result["AI 是未来"], 2)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|