update faults test

2019-07-21 02:09:23 +02:00 · 2019-07-21 02:09:23 +02:00 · 2061b3d763
parent 0470dd49f2
commit 2061b3d763
1 changed files with 6 additions and 1 deletions
--- a/tests/test_loader.py
+++ b/tests/test_loader.py
@ -104,7 +104,12 @@ class TestTTSDataset(unittest.TestCase):
                avg_length = mel_lengths.numpy().mean()
                assert avg_length >= last_length
            dataloader.dataset.sort_items()
-            assert frames[0] != dataloader.dataset.items[0]
+            is_items_reordered = False
+            for idx, item in enumerate(dataloader.dataset.items):
+                if item != frames[idx]:
+                    is_items_reordered = True
+                    break
+            assert is_items_reordered

    def test_padding_and_spec(self):
        if ok_ljspeech: