From c9f71978627844b7fe0803cec886ffa2129dbe9f Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Thu, 20 Jun 2024 14:16:45 +0200 Subject: [PATCH] test(helpers): add test_ prefix so tests actually run --- tests/tts_tests/test_helpers.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/tts_tests/test_helpers.py b/tests/tts_tests/test_helpers.py index 23bb440a..a83ec9dd 100644 --- a/tests/tts_tests/test_helpers.py +++ b/tests/tts_tests/test_helpers.py @@ -3,7 +3,7 @@ import torch as T from TTS.tts.utils.helpers import average_over_durations, generate_path, rand_segments, segment, sequence_mask -def average_over_durations_test(): # pylint: disable=no-self-use +def test_average_over_durations(): # pylint: disable=no-self-use pitch = T.rand(1, 1, 128) durations = T.randint(1, 5, (1, 21)) @@ -21,7 +21,7 @@ def average_over_durations_test(): # pylint: disable=no-self-use index += dur -def seqeunce_mask_test(): +def test_sequence_mask(): lengths = T.randint(10, 15, (8,)) mask = sequence_mask(lengths) for i in range(8): @@ -30,7 +30,7 @@ def seqeunce_mask_test(): assert mask[i, l:].sum() == 0 -def segment_test(): +def test_segment(): x = T.range(0, 11) x = x.repeat(8, 1).unsqueeze(1) segment_ids = T.randint(0, 7, (8,)) @@ -50,7 +50,7 @@ def segment_test(): assert x[idx, :, start_indx : start_indx + 10].sum() == segments[idx, :, :].sum() -def rand_segments_test(): +def test_rand_segments(): x = T.rand(2, 3, 4) x_lens = T.randint(3, 4, (2,)) segments, seg_idxs = rand_segments(x, x_lens, segment_size=3) @@ -68,7 +68,7 @@ def rand_segments_test(): assert all(x_lens_back == x_lens) -def generate_path_test(): +def test_generate_path(): durations = T.randint(1, 4, (10, 21)) x_length = T.randint(18, 22, (10,)) x_mask = sequence_mask(x_length).unsqueeze(1).long()