adjusting tests

sensein · Oct 1, 2024 · dbc68cd · dbc68cd
1 parent 5160191
commit dbc68cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tests/audio/tasks/features_extraction_test.py b/src/tests/audio/tasks/features_extraction_test.py
@@ -34,7 +34,7 @@ def test_extract_spectrogram_from_audios(resampled_mono_audio_sample: Audio) ->
     assert all(isinstance(spec["spectrogram"], torch.Tensor) for spec in result)
     # Spectrogram shape is (freq, time)
     assert all(spec["spectrogram"].dim() == 2 for spec in result)
-    assert all(spec["spectrogram"].shape[0] == 201 for spec in result)
+    assert all(spec["spectrogram"].shape[0] == 513 for spec in result)
 
 
 def test_extract_mel_spectrogram_from_audios(resampled_mono_audio_sample: Audio) -> None: