def test_fst_del_states():
fst = VectorFst()
# States
fst.add_state()
fst.add_state()
fst.delete_states()
assert fst.num_states() == 0
def test_state_iterator():
fst = VectorFst()
s1 = fst.add_state()
s2 = fst.add_state()
fst.set_start(s1)
fst.set_final(s2, 0.54)
tr1 = Tr(1, 18, 2.33, s2)
fst.add_tr(s1, tr1)
states_it = StateIterator(fst)
num_visited_states = 0
for idx, state in enumerate(states_it):
num_visited_states += 1
assert idx == state
assert num_visited_states == fst.num_states()