音量更新API関連テストの追加

Hiroshiba · Apr 28, 2024 · 1359299 · 1359299
1 parent 9923795
commit 1359299
Show file tree

Hide file tree

Showing 2 changed files with 126 additions and 0 deletions.
diff --git a/...napshots__/test_tts_engine/test_mocked_create_sing_volume_from_phoneme_and_f0_output.json b/...napshots__/test_tts_engine/test_mocked_create_sing_volume_from_phoneme_and_f0_output.json
diff --git a/test/tts_pipeline/test_tts_engine.py b/test/tts_pipeline/test_tts_engine.py
@@ -381,6 +381,26 @@ def test_mocked_synthesize_wave_output(snapshot_json: SnapshotAssertion) -> None
     assert snapshot_json == round_floats(result.tolist(), round_value=2)
 
 
+def test_mocked_create_sing_volume_from_phoneme_and_f0_output(
+    snapshot_json: SnapshotAssertion,
+) -> None:
+    """
+    モックされた `TTSEngine.create_sing_phoneme_and_f0_and_volume()` の出力スナップショットが一定である
+    """
+    # Inputs
+    tts_engine = TTSEngine(MockCoreWrapper())
+    doremi_srore = _gen_doremi_score()
+    phonemes, f0s, _ = tts_engine.create_sing_phoneme_and_f0_and_volume(
+        doremi_srore, StyleId(1)
+    )
+    # Outputs
+    result_volume = tts_engine.create_sing_volume_from_phoneme_and_f0(
+        doremi_srore, phonemes, f0s, StyleId(1)
+    )
+    # Tests
+    assert snapshot_json == round_floats(result_volume, round_value=2)
+
+
 def test_mocked_synthesize_wave_from_score_output(
     snapshot_json: SnapshotAssertion,
 ) -> None: