'''
对wordCloud中的函数进行测试python -m unittest
对于其中文件读取和绘制图片一类的函数没有进行测试
代码覆盖率测试coverage run -m unittest discover
'''

import unittest
from barrage import wordCloud as bw

class TestWordCloud(unittest.TestCase):
    word_list = [
        '盖 闻 天地 之 数 有 十二万 九千 六百 岁 为 一元',
        '将 一元 分为 十二 会 乃 子 丑 寅 卯 辰 巳 午 未 申 酉 戌 亥 之 十二支 也',
        '每 会 该 一万 八百岁',
        '天气 下降 地气 上升 天地 交合 群物 皆 生',
        '那座 山 正当 顶上 有 一块 仙石',
        '其 石 有 三丈 六尺 五寸 高 有 二丈 四尺 围圆 三丈 六尺 五寸 高 按 周天 三百六十五度 二丈 四尺 围圆 按 政历 二十四气',
        '盖 自 开辟 以来 每 受 天真 地 秀 日精 月华 感 之 既久 遂有 灵通 之 意',
        '内 育 仙胞 一日 迸裂 产 一 石 卵',
        '因 见风 化作 一个 石猴 五官 俱备 四肢 皆全 便就 学 爬 学 走 拜了 四方',
        '目 运 两道 金光 射 冲 斗府'
    ]

    def setUp(self):
        print('start')

    def test_ChangeDfToString(self):
        for i in range(10):
            self.assertTrue(type(bw.ChangeToFreq(self.word_list[i])) == dict)
        
    def tearDown(self):
        print('end')

if __name__ == '__main__':
    unittest.main()