admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fixed comments

Browse Source
This commit is contained in:
Tryndaron
2024-03-15 12:54:08 +01:00
parent 570048a2e0
commit fdb16f2f45
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scraibe/test/test_audio.py
+4 -6
View File
@@ -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)