Exemplo n.º 1
0
def test_fst_io_bytes():
    fst = VectorFst()

    # States
    s1 = fst.add_state()
    s2 = fst.add_state()

    fst.set_start(s1)
    fst.set_final(s2)

    fst_loaded = VectorFst.from_bytes(fst.to_bytes())

    assert fst_loaded == fst
Exemplo n.º 2
0
def test_fst_from_bytes():
    fst = VectorFst()

    # States
    s1 = fst.add_state()
    s2 = fst.add_state()

    fst.set_start(s1)
    fst.set_final(s2)

    with NamedTemporaryFile() as f:
        fst.write(f.name)
        bytes = Path(f.name).read_bytes()
        fst_loaded = VectorFst.from_bytes(bytes)

    assert fst == fst_loaded