From c5e44ecb183ba56917773e130116a7a06f2d3c22 Mon Sep 17 00:00:00 2001 From: Tryndaron Date: Mon, 8 Apr 2024 12:48:25 +0200 Subject: [PATCH] audio.py --- test/test_audio.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/test/test_audio.py b/test/test_audio.py index 9c39464..311a472 100644 --- a/test/test_audio.py +++ b/test/test_audio.py @@ -5,7 +5,7 @@ import torch DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") -TEST_WAVEFORM = torch.tensor([]).to(DEVICE) +TEST_WAVEFORM = torch.sin(torch.randn(160000)).to(DEVICE) TEST_SR = 16000 SAMPLE_RATE = 16000 NORMALIZATION_FACTOR = 32768 @@ -54,7 +54,7 @@ def test_AudioProcessor_init(probe_audio_processor): -def test_cut(): +def test_cut(probe_audio_processor): """Test the cut function of the AudioProcessor class. This test verifies that the cut function correctly extracts a segment of audio data from @@ -69,7 +69,11 @@ def test_cut(): start = 4 end = 7 - assert AudioProcessor(TEST_WAVEFORM, TEST_SR).cut(start, end).size() == int((end - start) * TEST_SR) + trimmed_waveform = probe_audio_processor.cut(start, end) + expected_size = int((end - start) * TEST_SR) + real_size = trimmed_waveform.size(0) + assert real_size == expected_size + #assert AudioProcessor(TEST_WAVEFORM, TEST_SR).cut(start, end).size() == int((end - start) * TEST_SR)