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
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