def test_phoneme_table_get_auxiliary_label_id(): phoneme_table = PhonemeTable() phoneme_table.add_label('a') phoneme_table.set_auxiliary_label('#0') phoneme_table.set_auxiliary_label('#1') assert phoneme_table.get_label_id('a') == 2 assert phoneme_table.get_auxiliary_label_id('#0') == 3 assert phoneme_table.get_auxiliary_label_id('#1') == 4
def test_phoneme_table_get_all_labels(): phoneme_table = PhonemeTable() phoneme_table.add_label('a') got = phoneme_table.get_all_labels() assert isinstance(got, dict) assert len(got) == 3 assert got[0] == '<epsilon>' assert got[1] == '<blank>' assert got[2] == 'a'
def test_phoneme_table_get_all_auxiliary_labels(): phoneme_table = PhonemeTable() phoneme_table.add_label('a') phoneme_table.set_auxiliary_label('#0') phoneme_table.set_auxiliary_label('#1') got = phoneme_table.get_all_auxiliary_labels() assert isinstance(got, dict) assert len(got) == 2 assert got[3] == '#0' assert got[4] == '#1'
def test_phoneme_table_add_label(): phoneme_table = PhonemeTable() phoneme_table.add_label('a') assert phoneme_table.num_labels() == 3 assert phoneme_table.get_label_id('a') == 2 assert phoneme_table.get_label(2) == 'a'