update tests

2026-09-11 12:00:24 +08:00 · 2020-07-08 10:21:45 +02:00
parent 6c60c182b5
commit 07d2d28ae6
6 changed files with 11 additions and 73 deletions
@@ -76,7 +76,7 @@ class TestTTSDataset(unittest.TestCase):
                # TODO: more assertion here
                assert type(speaker_name[0]) is str
                assert linear_input.shape[0] == c.batch_size
-                assert linear_input.shape[2] == self.ap.num_freq
+                assert linear_input.shape[2] == self.ap.fft_size // 2 + 1
                assert mel_input.shape[0] == c.batch_size
                assert mel_input.shape[2] == c.audio['num_mels']
                # check normalization ranges