roc/tests/test_boolean_algebra.rs

#[macro_use]
extern crate pretty_assertions;

extern crate bumpalo;
extern crate roc;

mod helpers;

#[cfg(test)]
mod test_boolean_algebra {
    use roc::subs::VarStore;
    use roc::uniqueness::boolean_algebra;
    use roc::uniqueness::boolean_algebra::BooleanAlgebra::{self, *};

    // HELPERS
    fn simplify_eq(mut a: BooleanAlgebra, mut b: BooleanAlgebra) {
        assert_eq!(a.simplify(), b.simplify());
    }

    #[test]
    fn true_in_or() {
        let var_store = VarStore::default();
        let var = var_store.fresh();

        simplify_eq(
            Disjunction(Box::new(Ground(true)), Box::new(Variable(var))),
            Ground(true),
        );
    }

    #[test]
    fn false_in_or() {
        let var_store = VarStore::default();
        let var = var_store.fresh();

        simplify_eq(
            Disjunction(Box::new(Ground(false)), Box::new(Variable(var))),
            Variable(var),
        );
    }

    #[test]
    fn false_in_and() {
        let var_store = VarStore::default();
        let var = var_store.fresh();

        simplify_eq(
            Conjunction(Box::new(Ground(false)), Box::new(Variable(var))),
            Ground(false),
        );
    }

    #[test]
    fn unify_single_var() {
        let var_store = VarStore::default();
        let var = var_store.fresh();

        let result = boolean_algebra::unify(&Variable(var), &Ground(true));

        if let Some(sub) = result {
            assert_eq!(Some(&Ground(false)), sub.get(var));
        } else {
            panic!("result is None");
        }
    }

    #[test]
    fn unify_or() {
        let var_store = VarStore::default();
        let a = var_store.fresh();
        let b = var_store.fresh();

        let result = boolean_algebra::unify(
            &Disjunction(Box::new(Variable(a)), Box::new(Variable(b))),
            &Ground(true),
        );

        if let Some(sub) = result {
            assert_eq!(Some(&Variable(b)), sub.get(a));
            assert_eq!(Some(&Ground(false)), sub.get(b));
        } else {
            panic!("result is None");
        }
    }
}