class TestSpeech(unittest.TestCase): def setUp(self): self.dataset = SpeechDataset(flist='train.flist', aligns='aligns.pdf', template_shell_command='copy-feats-1file', subset=100, preprocessor=None) def test_io(self): "Tests that the data is loaded correctly" iterator = self.dataset.iterator() for batch in iterator: pass
def setUp(self): self.dataset = SpeechDataset(flist='train.flist', aligns='aligns.pdf', template_shell_command='copy-feats-1file', subset=100, preprocessor=None)