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()