Ethanfel/ComfyUI-Omnivoice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update tests for chapter_title output and guidance_scale param

Browse Source 
- epub loader tests: unpack 3 return values (added chapter_title)
- generator tests: expect guidance_scale in model.generate() calls

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-04-09 19:06:38 +02:00
parent 0f2bcc4c0e
commit f8657aca80
2 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_generator.py
+2 -2
View File
@@ -46,7 +46,7 @@ def test_generate_auto_voice():
assert "sample_rate" in audio
assert audio["sample_rate"] == 24000
mock_model.generate.assert_called_once_with(
text="Hello world", speed=1.0, num_step=32
text="Hello world", speed=1.0, num_step=32, guidance_scale=2.0
)
@@ -64,7 +64,7 @@ def test_generate_voice_design():
audio = result[0]
assert audio["sample_rate"] == 24000
mock_model.generate.assert_called_once_with(
text="Hello world", instruct="female, low pitch", speed=1.0, num_step=32
text="Hello world", instruct="female, low pitch", speed=1.0, num_step=32, guidance_scale=2.0
)