test: generate cover (#282)

* test: generate cover

* docs: update docs

Author: timerring

README.md

@@ -173,7 +173,7 @@ pip install -r requirements.txt
 > [!TIP]
 > - 有关自动切片的配置在 `bilive.toml` 文件的 `[slice]` 部分。
 > - `auto_slice` 默认为 false, 即不进行自动切片。
-> - 可以通过单元测试调试你自己的 prompt，单元测试在 `tests/test_autoslice.py`，执行 `python -m unittest` 即可，后接 `tests.test_autoslice` 测试整个模块，`tests.test_autoslice.TestXXXMain` 测试某个模型。
+> - 可以通过单元测试调试你自己的 prompt，单元测试在 `tests/test_autoslice.py`，执行 `python -m unittest` 即可，后接 `tests.test_autoslice` 测试整个模块，`tests.test_autoslice.TestXXXMain` 测试某个模型。部分模型会返回多个标题，请在 prompt 中指出，仅返回一个标题的字符串即可。
 
 MLLM 模型主要用于自动切片后的切片标题生成，此功能默认关闭，如果需要打开请将 `auto_slice` 参数设置为 `true`，并且写下你自己的 `slice_prompt`(可以包含 `{artist}` 关键词会自动替换)，其他配置分别有：
 - `slice_duration` 以秒为单位设置切片时长（不建议超过 180 秒）。
@@ -196,6 +196,7 @@ MLLM 模型主要用于自动切片后的切片标题生成，此功能默认关
 > [!TIP]
 > - 有关自动生成视频封面的配置在 `bilive.toml` 文件的 `[cover]` 部分。
 > - `generate_cover` 默认为 false, 即不进行自动生成视频封面。
+> - 可以通过单元测试调试你自己的 prompt，单元测试在 `tests/test_cover.py`，执行 `python -m unittest` 即可，后接 `tests.test_cover` 测试整个模块，`tests.test_cover.TestXXXMain` 测试某个模型。
 
 采用图生图多模态模型，自动获取视频截图并上传风格变换后的视频封面，如需使用本功能，请将 `generate_cover` 参数设置为 `true`，并且写下你自己的 prompt，注意部分模型只支持英文，接下来需要配置的参数有 image_gen_model 和对应的 api key，请自行根据链接注册账号并且申请对应 api key，填写在对应的参数中，请注意以下模型只有你在 `image_gen_model` 参数中设置的那个模型会生效。
 

tests/test_cover.py

@@ -0,0 +1,80 @@
+import unittest
+from src.cover.image_model_sdk.amazon_sdk import amazon_generate_cover
+from src.cover.image_model_sdk.baidu_sdk import baidu_generate_cover
+from src.cover.image_model_sdk.hidream_sdk import hidream_generate_cover
+from src.cover.image_model_sdk.ideogram_sdk import ideogram_generate_cover
+from src.cover.image_model_sdk.kolors_sdk import kolors_generate_cover
+from src.cover.image_model_sdk.luma_sdk import luma_generate_cover
+from src.cover.image_model_sdk.minimax_sdk import minimax_generate_cover
+from src.cover.image_model_sdk.recraft_sdk import recraft_generate_cover
+from src.cover.image_model_sdk.stability_sdk import stable_diffusion_generate_cover
+from src.cover.image_model_sdk.tencent_sdk import hunyuan_generate_cover
+
+
+class BaseTest(unittest.TestCase):
+    file_path = "your_image_path"
+
+
+class TestAmazonMain(BaseTest):
+    def test_amazon_generate_cover(self):
+        cover_path = amazon_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+
+class TestBaiduMain(BaseTest):
+    def test_baidu_generate_cover(self):
+        cover_path = baidu_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestHidreamMain(BaseTest):
+    def test_hidream_generate_cover(self):
+        cover_path = hidream_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestIdeogramMain(BaseTest):
+    def test_ideogram_generate_cover(self):
+        cover_path = ideogram_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestKolorsMain(BaseTest):
+    def test_kolors_generate_cover(self):
+        cover_path = kolors_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestLumaMain(BaseTest):
+    def test_luma_generate_cover(self):
+        cover_path = luma_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestMinimaxMain(BaseTest):
+    def test_minimax_generate_cover(self):
+        cover_path = minimax_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestRecraftMain(BaseTest):
+    def test_recraft_generate_cover(self):
+        cover_path = recraft_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestStabilityMain(BaseTest):
+    def test_stable_diffusion_generate_cover(self):
+        cover_path = stable_diffusion_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+class TestTencentMain(BaseTest):
+    def test_hunyuan_generate_cover(self):
+        cover_path = hunyuan_generate_cover(self.file_path)
+        self.assertIsNotNone(cover_path)
+        print(cover_path, flush=True)
+
+if __name__ == "__main__":
+    unittest.main()