整理: e2e single API テスト vol 8

test/e2e/single_api/__snapshots__/test_frame_synthesis.ambr

@@ -0,0 +1,4 @@
+# serializer version: 1
+# name: test_post_frame_synthesis_200
+  'MD5:ee67a7e22f9e6c2ec519796610a2ed21'
+# ---

test/e2e/single_api/__snapshots__/test_synthesis.ambr

@@ -0,0 +1,4 @@
+# serializer version: 1
+# name: test_post_synthesis_200
+  'MD5:c337b99b27bc5bc461faa58df029b302'
+# ---

test/e2e/single_api/test_audio_query.py

@@ -1,5 +1,5 @@
 """
-AudioQuery APIのテスト
+/audio_query API のテスト
 """
 
 from test.utility import round_floats
@@ -8,7 +8,7 @@
 from syrupy.assertion import SnapshotAssertion
 
 
-def test_speakerを指定して音声合成クエリが取得できる(
+def test_post_audio_query_200(
     client: TestClient, snapshot_json: SnapshotAssertion
 ) -> None:
     response = client.post("/audio_query", params={"text": "テストです", "speaker": 0})

test/e2e/single_api/test_frame_synthesis.py

@@ -2,10 +2,17 @@
 /frame_synthesis API のテスト
 """
 
+import io
+from test.utility import hash_long_string, round_floats
+
+import soundfile as sf
 from fastapi.testclient import TestClient
+from syrupy.assertion import SnapshotAssertion
 
 
-def test_post_frame_synthesis_200(client: TestClient) -> None:
+def test_post_frame_synthesis_200(
+    client: TestClient, snapshot: SnapshotAssertion
+) -> None:
     query = {
         "f0": [
             0.0,
@@ -81,3 +88,11 @@ def test_post_frame_synthesis_200(client: TestClient) -> None:
     }
     response = client.post("/frame_synthesis", params={"speaker": 0}, json=query)
     assert response.status_code == 200
+
+    # FileResponse 内の .wav から抽出された音声波形が一致する
+    assert response.headers["content-type"] == "audio/wav"
+    wave = sf.read(io.BytesIO(response.read()))[0].tolist()
+    # NOTE: Linux-Windows 数値精度問題に対するワークアラウンド
+    wave = round_floats(wave, 2)
+    wave_str = " ".join(map(lambda point: str(point), wave))
+    assert snapshot == hash_long_string(wave_str)

test/e2e/single_api/test_multi_synthesis.py

@@ -59,3 +59,17 @@ def test_post_multi_synthesis_200(client: TestClient) -> None:
     ]
     response = client.post("/multi_synthesis", params={"speaker": 0}, json=queries)
     assert response.status_code == 200
+
+    # FIXME: ZIP ファイル内の .wav に Linux-Windows 数値精度問題があるため、スナップショットテストには解凍が必要
+    assert response.headers["content-type"] == "application/zip"
+    # import io
+    # from test.utility import hash_long_string, round_floats
+    # import soundfile as sf
+    # from syrupy.assertion import SnapshotAssertion
+    # FileResponse 内の zip ファイルに圧縮された .wav から抽出された音声波形が一致する
+    # # zip 解凍
+    # waves = concatenate_func(map(lambda path: sf.read(path)[0].tolist(), wave_paths))
+    # # NOTE: Linux-Windows 数値精度問題に対するワークアラウンド
+    # waves = round_floats(waves, 2)
+    # waves_str = " ".join(map(lambda point: str(point), waves))
+    # assert snapshot == hash_long_string(waves_str)

test/e2e/single_api/test_synthesis.py

@@ -2,12 +2,16 @@
 /synthesis API のテスト
 """
 
+import io
 from test.e2e.single_api.utils import gen_mora
+from test.utility import hash_long_string, round_floats
 
+import soundfile as sf
 from fastapi.testclient import TestClient
+from syrupy.assertion import SnapshotAssertion
 
 
-def test_post_synthesis_200(client: TestClient) -> None:
+def test_post_synthesis_200(client: TestClient, snapshot: SnapshotAssertion) -> None:
     query = {
         "accent_phrases": [
             {
@@ -33,3 +37,11 @@ def test_post_synthesis_200(client: TestClient) -> None:
     }
     response = client.post("/synthesis", params={"speaker": 0}, json=query)
     assert response.status_code == 200
+
+    # FileResponse 内の .wav から抽出された音声波形が一致する
+    assert response.headers["content-type"] == "audio/wav"
+    wave = sf.read(io.BytesIO(response.read()))[0].tolist()
+    # NOTE: Linux-Windows 数値精度問題に対するワークアラウンド
+    wave = round_floats(wave, 2)
+    wave_str = " ".join(map(lambda point: str(point), wave))
+    assert snapshot == hash_long_string(wave_str)

test/e2e/single_api/test_synthesis_morphing.py

@@ -37,3 +37,16 @@ def test_post_synthesis_morphing_200(client: TestClient) -> None:
         json=queries,
     )
     assert response.status_code == 200
+
+    # FIXME: Linux-MacOS 数値精度問題によりスナップショットテストがコケる
+    # import io
+    # from test.utility import hash_long_string, round_floats
+    # import soundfile as sf
+    # from syrupy.assertion import SnapshotAssertion
+    # # FileResponse 内の .wav から抽出された音声波形が一致する
+    # assert response.headers["content-type"] == "audio/wav"
+    # wave = sf.read(io.BytesIO(response.read()))[0].tolist()
+    # # NOTE: Linux-Windows 数値精度問題に対するワークアラウンド
+    # wave = round_floats(wave, 2)
+    # wave_str = " ".join(map(lambda point: str(point), wave))
+    # assert snapshot == hash_long_string(wave_str)