def test_dump_and_load(self):
''' test dump and load data '''
temp_dir = self.get_temp_dir()
num_spk = 5
num_utt_per_spk = 3
meta = gen_dummy_meta(num_spk, num_utt_per_spk)
meta.dump(temp_dir, True)
with open(os.path.join(temp_dir, 'feats.scp'), 'r') as fp_in:
logging.info('feats.scp:\n%s' % (fp_in.read()))
loaded_meta = kaldi_dir.KaldiMetaData()
loaded_meta.load(temp_dir)
self.assertEqual(len(meta.utts), len(loaded_meta.utts))
for utt_key in meta.utts.keys():
self.assertIn(utt_key, loaded_meta.utts)
self.assertEqual(len(meta.spks), len(loaded_meta.spks))
for spk_key in meta.spks.keys():
self.assertIn(spk_key, loaded_meta.spks)
def test_gen_dummy_data(self):
''' test dump and load data '''
num_spk = 5
num_utt_per_spk = 3
meta = gen_dummy_meta(num_spk, num_utt_per_spk)
self.assertEqual(len(meta.spks), num_spk)