add test

huggingface · Feb 2, 2022 · e8ebbaa · e8ebbaa · github-actions · Feb 2, 2022
1 parent d9e7618
commit e8ebbaa
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tests/features/data/test_audio_16000.mp3 b/tests/features/data/test_audio_16000.mp3
diff --git a/tests/features/test_audio.py b/tests/features/test_audio.py
@@ -131,6 +131,25 @@ def test_audio_decode_example_mp3(shared_datadir):
     assert decoded_example["path"] == audio_path
     assert decoded_example["array"].shape == (109440,)
     assert decoded_example["sampling_rate"] == 44100
+
+@require_sox
+@require_torchaudio
+def test_audio_resampling_mp3_different_sampling_rates(shared_datadir):
+    audio_path = str(shared_datadir / "test_audio_44100.mp3")
+    audio_path2 = str(shared_datadir / "test_audio_16000.mp3")
+    audio = Audio(sampling_rate=48000)
+
+    decoded_example = audio.decode_example(audio.encode_example(audio_path))
+    assert decoded_example.keys() == {"path", "array", "sampling_rate"}
+    assert decoded_example["path"] == audio_path
+    assert decoded_example["array"].shape == (119119,)
+    assert decoded_example["sampling_rate"] == 48000
+
+    decoded_example = audio.decode_example(audio.encode_example(audio_path2))
+    assert decoded_example.keys() == {"path", "array", "sampling_rate"}
+    assert decoded_example["path"] == audio_path
+    assert decoded_example["array"].shape == (120960,)
+    assert decoded_example["sampling_rate"] == 48000
 
 
 @require_sndfile