feat(wordcloud): add wordcloud_chineseize (#15)

* feat(wordcloud): add wordcloud_chineseize

CHANGELOG.md

@@ -3,6 +3,10 @@
 
 ## [Unreleased]
 
+### Added
+- Support chinese display in `wordcloud`
+
+
 ## [0.1.1] - 2024-12-01
 
 ### Changed

README.md

@@ -39,6 +39,10 @@ uv add zhplot
 import matplotlib.pyplot as plt
 ```
 
+### 支持的框架
+- [x] [matplotlib](https://github.com/matplotlib/matplotlib)
+- [x] [wordcloud](https://github.com/amueller/word_cloud)
+
 ### 一个简单的例子
 <div align="center">
     <img src="https://github.com/shenxiangzhuang/zhplot/blob/7569552e07a8b4de7afd8c9df5cbcb154a349e97/docs/images/zhplot_demo.png?raw=true" width="500"/>
@@ -56,6 +60,7 @@ plt.ylabel('纵坐标')
 plt.show()
 ```
 
+
 ## 相似项目
 - [japanize-matplotlib](https://github.com/uehara1414/japanize-matplotlib)
 - [chineseize-matplotlib](https://github.com/cndeng/chineseize-matplotlib)

tests/test_wordcloud.py

@@ -0,0 +1,10 @@
+import os
+
+import zhplot  # noqa
+
+
+def test_chinese_font_config():
+    """Test if wordcloud is configured to use Chinese fonts."""
+    font_path_env = os.getenv("FONT_PATH")
+    assert font_path_env is not None
+    assert "NotoSansSC-Regular" in font_path_env

zhplot/__init__.py

@@ -1,5 +1,7 @@
 from .matplotlib import matplotlib_chineseize
+from .wordcloud import wordcloud_chineseize
 
 __all__ = [
     "matplotlib_chineseize",
+    "wordcloud_chineseize",
 ]

zhplot/wordcloud.py

@@ -0,0 +1,10 @@
+import os
+from pathlib import Path
+
+
+def wordcloud_chineseize():
+    font_ttf: str = "NotoSansSC-Regular.ttf"
+    os.environ["FONT_PATH"] = f"{Path(__file__).parent}/fonts/{font_ttf}"
+
+
+wordcloud_chineseize()