From 9ef2c7ed624fda8ac8052ea3824132b5ab6b4481 Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Mon, 2 Dec 2024 00:09:39 +0100 Subject: [PATCH] test(freevc): fix output length check --- tests/vc_tests/test_freevc.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/vc_tests/test_freevc.py b/tests/vc_tests/test_freevc.py index 914237b5..fe07b272 100644 --- a/tests/vc_tests/test_freevc.py +++ b/tests/vc_tests/test_freevc.py @@ -31,7 +31,7 @@ class TestFreeVC(unittest.TestCase): @staticmethod def _create_inputs_inference(): - source_wav = torch.rand(16000) + source_wav = torch.rand(15999) target_wav = torch.rand(16000) return source_wav, target_wav @@ -96,8 +96,8 @@ class TestFreeVC(unittest.TestCase): source_wav, target_wav = self._create_inputs_inference() output_wav = model.voice_conversion(source_wav, target_wav) assert ( - output_wav.shape[0] + config.audio.hop_length == source_wav.shape[0] - ), f"{output_wav.shape} != {source_wav.shape}" + output_wav.shape[0] == source_wav.shape[0] - source_wav.shape[0] % config.audio.hop_length + ), f"{output_wav.shape} != {source_wav.shape}, {config.audio.hop_length}" def test_train_step(self): ...