use std::fmt::Display;
use anyhow::Result;
use crate::algorithms::fst_convert_from_ref;
use crate::algorithms::rm_epsilon::{rm_epsilon, RmEpsilonFst};
use crate::fst_impls::VectorFst;
use crate::fst_properties::FstProperties;
use crate::fst_traits::{MutableFst, SerializableFst};
use crate::semirings::WeaklyDivisibleSemiring;
use crate::semirings::{SerializableSemiring, WeightQuantize};
use crate::tests_openfst::utils::test_eq_fst;
use crate::tests_openfst::FstTestData;
pub fn test_rmepsilon<W, F>(test_data: &FstTestData<W, F>) -> Result<()>
where
F: SerializableFst<W> + MutableFst<W> + Display,
W: SerializableSemiring + WeaklyDivisibleSemiring + WeightQuantize,
{
let mut fst_rmepsilon = test_data.raw.clone();
rm_epsilon(&mut fst_rmepsilon)?;
std::dbg!(fst_rmepsilon.properties());
std::dbg!(test_data.rmepsilon.result_static.properties());
assert!(fst_rmepsilon
.properties_check(FstProperties::NO_EPSILONS)?
.contains(FstProperties::NO_EPSILONS));
test_eq_fst(
&test_data.rmepsilon.result_static,
&fst_rmepsilon,
"RmEpsilon",
);
Ok(())
}
pub fn test_rmepsilon_lazy<W>(test_data: &FstTestData<W, VectorFst<W>>) -> Result<()>
where
W: SerializableSemiring + WeightQuantize,
{
let rmepsilon_lazy_fst_openfst = &test_data.rmepsilon.result_lazy;
let rmepsilon_lazy_fst: VectorFst<_> =
fst_convert_from_ref(&RmEpsilonFst::new(test_data.raw.clone())?);
test_eq_fst(
rmepsilon_lazy_fst_openfst,
&rmepsilon_lazy_fst,
"RmEpsilon lazy",
);
Ok(())
}