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.
27 lines
779 B
27 lines
779 B
import unittest
|
|
import a_wordcloud
|
|
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 TestAWordcloud(unittest.TestCase):
|
|
|
|
def test_blue_color_func(self):
|
|
color = a_wordcloud.blue_color_func()
|
|
self.assertTrue(color.startswith("hsl(210, 100%,"))
|
|
|
|
def test_wordcloud_generation(self):
|
|
dm = {
|
|
'danmu': ["测试弹幕"] * 10
|
|
}
|
|
with self.assertRaises(Exception):
|
|
a_wordcloud.wordcloud_generation(dm)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|