def test_small_fst():
fst = VectorFst()
# States
s1 = fst.add_state()
s2 = fst.add_state()
assert s1 == 0
assert s2 == 1
fst.set_start(s1)
fst.set_final(s2)
assert fst.start() == s1
assert fst.is_final(s2)
assert pytest.approx(fst.final(s2)) == pytest.approx(0.0)
# Trs
tr_1 = Tr(3, 5, 10.0, s2)
fst.add_tr(s1, tr_1)
assert fst.num_trs(s1) == 1
tr_2 = Tr(5, 7, 18.0, s2)
fst.add_tr(s1, tr_2)
assert fst.num_trs(s1) == 2
def test_final_weight():
fst = VectorFst()
s1 = fst.add_state()
fst.set_final(s1, 2.3)
assert pytest.approx(fst.final(s1)) == pytest.approx(2.3)
