def setUp(self):
self.utterance = Utterance("utt_id_1", "record_1", None)
self.ctm_record = CtmRecord(["duration", "start_time"], "id1")
self.ctm_record.add_rows([[1, 0], [1, 2], [1, 3]])
self.hypo_storage = HypothesisRecordStorage.storage
self.hypo_storage["record_1"] = self.ctm_record
self.pool = UnmarkedWordPool()
class TestUnlabeledWordPool(TestCase):
def setUp(self):
self.utterance = Utterance("utt_id_1", "record_1", None)
self.ctm_record = CtmRecord(["duration", "start_time"], "id1")
self.ctm_record.add_rows([[1, 0], [1, 2], [1, 3]])
self.hypo_storage = HypothesisRecordStorage.storage
self.hypo_storage["record_1"] = self.ctm_record
self.pool = UnmarkedWordPool()
def test_add_utterances(self):
self.pool.add_utterances([self.utterance])
self.assertEqual(self.pool[0], self.utterance)
def test_iter(self):
self.pool.add_utterances([self.utterance])
expected_seq = [(i, line_record) for (i, line_record) in enumerate(self.ctm_record)]
pool_sequence = [x for x in self.pool]
self.assertSequenceEqual(pool_sequence, expected_seq)
def setup_hypo_storage(self):
ctm_record = CtmRecord(["duration", "start_time"], "id1")
ctm_record.add_rows([[1, 0], [1, 2], [1, 3]])
self.hypo_storage = HypothesisRecordStorage.storage
self.hypo_storage["record_1"] = ctm_record
