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)