From fdb16f2f45cb79d574d3bc0ba850c76b3b7d5352 Mon Sep 17 00:00:00 2001 From: Tryndaron Date: Fri, 15 Mar 2024 12:54:08 +0100 Subject: [PATCH] fixed comments --- scraibe/test/test_audio.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/scraibe/test/test_audio.py b/scraibe/test/test_audio.py index 5a35055..2a0ce53 100644 --- a/scraibe/test/test_audio.py +++ b/scraibe/test/test_audio.py @@ -3,10 +3,9 @@ from .audio import AudioProcessor import torch -device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') - -test_waveform = torch.tensor([]).to(device) +device = torch.device("cuda" if torch.cuda.is_available() else "cpu") +TEST_WAVEFORM = torch.tensor([]).to(device) TEST_SR = 16000 SAMPLE_RATE = 16000 NORMALIZATION_FACTOR = 32768 @@ -43,11 +42,10 @@ def test_AudioProcessor_init(probe_audio_processor): def test_cut(): """Test for the test_cut Method for fixed parameters """ - waveform = torch.Tensor(10, 3) - sr = 16000 + start = 4 end = 7 - assert AudioProcessor(waveform, sr).cut(start, end).size() == int((end - start) * TEST_SR) + assert AudioProcessor(TEST_WAVEFORM, TEST_SR).cut(start, end).size() == int((end - start) * TEST_SR)