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'