rustfst 1.3.0

Library for constructing, combining, optimizing, and searching weighted finite-state transducers (FSTs).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use std::sync::Arc;

use anyhow::Result;

use crate::algorithms::fst_convert_from_ref;
use crate::fst_impls::{ConstFst, VectorFst};
use crate::fst_traits::Fst;
use crate::semirings::{SerializableSemiring, WeightQuantize};
use crate::tests_openfst::utils::test_eq_fst;
use crate::tests_openfst::FstTestData;
use crate::SymbolTable;

pub fn test_const_fst_convert_convert<W>(test_data: &FstTestData<W, VectorFst<W>>) -> Result<()>
where
    W: SerializableSemiring + WeightQuantize,
{
    let mut raw_fst = test_data.raw.clone();

    {
        let mut isymt = SymbolTable::new();
        isymt.add_symbol("a");
        let mut osymt = SymbolTable::new();
        osymt.add_symbol("b");
        osymt.add_symbol("c");

        raw_fst.set_input_symbols(Arc::new(isymt));
        raw_fst.set_output_symbols(Arc::new(osymt));
    }

    let const_fst: ConstFst<_> = raw_fst.clone().into();
    let pred_fst: VectorFst<_> = fst_convert_from_ref(&const_fst);

    test_eq_fst(
        &raw_fst,
        &pred_fst,
        "Convert VectorFst -> ConstFst -> VectorFst",
    );

    // Check symbol tables are still attached
    {
        let isymt = const_fst.input_symbols();
        assert!(isymt.is_some());
        let isymt = isymt.unwrap();
        assert_eq!(isymt.len(), 2);

        let osymt = const_fst.output_symbols();
        assert!(osymt.is_some());
        let osymt = osymt.unwrap();
        assert_eq!(osymt.len(), 3);
    }
    {
        let isymt = pred_fst.input_symbols();
        assert!(isymt.is_some());
        let isymt = isymt.unwrap();
        assert_eq!(isymt.len(), 2);

        let osymt = pred_fst.output_symbols();
        assert!(osymt.is_some());
        let osymt = osymt.unwrap();
        assert_eq!(osymt.len(), 3);
    }
    Ok(())
}