use nom:: { IResult, branch::alt, bytes::complete::{escaped_transform, tag}, character::complete::{digit1, multispace0, none_of}, combinator::{map, map_res, opt, recognize}, multi::{many1, separated_list}, sequence::{delimited, pair, preceded, terminated}, }; use super::{Declarations, function_or_predicate_name, word_boundary, variable_name}; fn negative<'i>(i: &'i str, d: &Declarations) -> IResult<&'i str, crate::Term> { map ( preceded ( terminated ( tag("-"), multispace0, ), |i| term_precedence_1(i, d), ), |x| match x { crate::Term::Integer(value) => crate::Term::integer(-value), crate::Term::UnaryOperation( crate::UnaryOperation{operator: crate::UnaryOperator::Negative, argument}) => *argument, _ => crate::Term::negative(Box::new(x)), } )(i) } fn absolute_value<'i>(i: &'i str, d: &Declarations) -> IResult<&'i str, crate::Term> { map ( delimited ( terminated ( tag("|"), multispace0, ), |i| term(i, d), preceded ( multispace0, tag("|"), ), ), |x| crate::Term::absolute_value(Box::new(x)), )(i) } fn true_(i: &str) -> IResult<&str, crate::Term> { map ( terminated ( tag("true"), word_boundary, ), |_| crate::Term::true_(), )(i) } fn false_(i: &str) -> IResult<&str, crate::Term> { map ( terminated ( tag("false"), word_boundary, ), |_| crate::Term::false_(), )(i) } pub fn boolean(i: &str) -> IResult<&str, crate::Term> { alt (( true_, false_, ))(i) } pub fn integer(i: &str) -> IResult<&str, crate::Term> { map ( map_res ( recognize ( terminated ( pair ( opt ( alt (( tag("-"), tag("+"), )) ), digit1, ), word_boundary, ) ), std::str::FromStr::from_str, ), crate::Term::integer, )(i) } fn infimum(i: &str) -> IResult<&str, crate::Term> { map ( terminated ( tag("#inf"), word_boundary, ), |_| crate::Term::infimum(), )(i) } fn supremum(i: &str) -> IResult<&str, crate::Term> { map ( terminated ( tag("#sup"), word_boundary, ), |_| crate::Term::supremum(), )(i) } pub fn special_integer(i: &str) -> IResult<&str, crate::Term> { alt (( infimum, supremum, ))(i) } pub fn string(i: &str) -> IResult<&str, crate::Term> { map ( terminated ( delimited ( tag("\""), escaped_transform ( none_of("\"\\"), '\\', alt (( tag("\""), tag("\\"), map ( tag("n"), |_| "\n", ), map ( tag("t"), |_| "\t", ), )), ), tag("\""), ), word_boundary, ), crate::Term::string, )(i) } pub(crate) fn function_or_predicate<'i>(i: &'i str, d: &Declarations) -> IResult<&'i str, (&'i str, Option>>)> { pair ( function_or_predicate_name, opt ( delimited ( delimited ( multispace0, tag("("), multispace0, ), separated_list ( delimited ( multispace0, tag(","), multispace0, ), map ( |i| term(i, d), Box::new, ), ), preceded ( multispace0, tag(")"), ), ) ), )(i) } pub fn function<'i>(i: &'i str, d: &Declarations) -> IResult<&'i str, crate::Function> { map ( |i| function_or_predicate(i, d), |(name, arguments)| { let arguments = match arguments { Some(arguments) => arguments, None => vec![], }; let mut function_declarations = d.function_declarations.borrow_mut(); let declaration = match function_declarations.iter() .find(|x| x.name == name && x.arity == arguments.len()) { Some(declaration) => std::rc::Rc::clone(&declaration), None => { let declaration = crate::FunctionDeclaration { name: name.to_string(), arity: arguments.len(), }; let declaration = std::rc::Rc::new(declaration); function_declarations.insert(std::rc::Rc::clone(&declaration)); declaration }, }; crate::Function::new(declaration, arguments) }, )(i) } pub fn variable_declaration(i: &str) -> IResult<&str, crate::VariableDeclaration> { map ( variable_name, |name| crate::VariableDeclaration::new(name.to_string()) )(i) } pub fn variable<'i>(i: &'i str, d: &Declarations) -> IResult<&'i str, crate::Variable> { map ( variable_name, |name| { let mut variable_declaration_stack = d.variable_declaration_stack.borrow_mut(); let declaration = variable_declaration_stack.find_or_create(name); crate::Variable::new(declaration) }, )(i) } fn term_parenthesized<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { delimited ( terminated ( tag("("), multispace0, ), |i| term(i, d), preceded ( multispace0, tag(")"), ), )(i) } fn term_precedence_0<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { alt (( boolean, special_integer, integer, map ( |i| function(i, d), crate::Term::Function, ), string, map ( |i| variable(i, d), crate::Term::Variable, ), |i| absolute_value(i, d), |i| term_parenthesized(i, d), ))(i) } fn term_precedence_1<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { alt (( |i| negative(i, d), |i| term_precedence_0(i, d), ))(i) } fn term_precedence_2<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { alt (( map ( pair ( many1 ( terminated ( |i| term_precedence_1(i, d), delimited ( multispace0, tag("**"), multispace0, ), ) ), |i| term_precedence_1(i, d), ), |(arguments, last_argument)| arguments.into_iter().rev().fold(last_argument, |accumulator, argument| crate::Term::exponentiate(Box::new(argument), Box::new(accumulator))), ), |i| term_precedence_1(i, d), ))(i) } fn term_precedence_3<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { alt (( map ( pair ( |i| term_precedence_2(i, d), many1 ( pair ( delimited ( multispace0, alt (( tag("*"), tag("/"), tag("%"), )), multispace0, ), |i| term_precedence_2(i, d), ) ), ), |(first_argument, arguments)| arguments.into_iter().fold(first_argument, |accumulator, (operator, argument)| match operator { "*" => crate::Term::multiply(Box::new(accumulator), Box::new(argument)), "/" => crate::Term::divide(Box::new(accumulator), Box::new(argument)), "%" => crate::Term::modulo(Box::new(accumulator), Box::new(argument)), // TODO: handle appropriately _ => panic!("test"), }) ), |i| term_precedence_2(i, d), ))(i) } fn term_precedence_4<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { alt (( map ( pair ( |i| term_precedence_3(i, d), many1 ( pair ( delimited ( multispace0, alt (( tag("+"), tag("-"), )), multispace0, ), |i| term_precedence_3(i, d), ) ), ), |(first_argument, arguments)| arguments.into_iter().fold(first_argument, |accumulator, (operator, argument)| match operator { "+" => crate::Term::add(Box::new(accumulator), Box::new(argument)), "-" => crate::Term::subtract(Box::new(accumulator), Box::new(argument)), // TODO: handle appropriately _ => panic!("test"), }) ), |i| term_precedence_3(i, d), ))(i) } pub fn term<'a>(i: &'a str, d: &Declarations) -> IResult<&'a str, crate::Term> { term_precedence_4(i, d) } #[cfg(test)] mod tests { use crate::parse::terms::*; use crate::{Term, VariableDeclaration, VariableDeclarationStack}; fn term(i: &str) -> Term { crate::parse::term(i, &Declarations::new()).unwrap().1 } fn format_term(i: &str) -> String { format!("{}", term(i)) } #[test] fn parse_term_parenthesized() { assert_eq!(format_term("(1)"), format_term("1")); assert_eq!(format_term("((1))"), format_term("1")); assert_eq!(format_term("(-1)"), format_term("-1")); assert_eq!(format_term("((-1))"), format_term("-1")); assert_eq!(format_term("(-(1))"), format_term("-1")); assert_eq!(format_term("-((1))"), format_term("-1")); assert_eq!(format_term("(-(-1))"), format_term("1")); assert_eq!(format_term("-((-1))"), format_term("1")); assert_eq!(format_term("-(-(1))"), format_term("1")); assert_eq!(format_term("-(-(-1))"), format_term("-1")); assert_eq!(format_term("(a)"), format_term("a")); assert_eq!(format_term("((a))"), format_term("a")); assert_eq!(format_term("(X)"), format_term("X")); assert_eq!(format_term("((X))"), format_term("X")); assert_eq!(format_term("(\"test\")"), format_term("\"test\"")); assert_eq!(format_term("((\"test\"))"), format_term("\"test\"")); assert_eq!(format_term("(a ** b)"), format_term("a ** b")); assert_eq!(format_term("(a * b)"), format_term("a * b")); assert_eq!(format_term("(a / b)"), format_term("a / b")); assert_eq!(format_term("(a % b)"), format_term("a % b")); assert_eq!(format_term("(a + b)"), format_term("a + b")); assert_eq!(format_term("(a - b)"), format_term("a - b")); assert_eq!(format_term("((a ** b))"), format_term("a ** b")); assert_eq!(format_term("((a * b))"), format_term("a * b")); assert_eq!(format_term("((a / b))"), format_term("a / b")); assert_eq!(format_term("((a % b))"), format_term("a % b")); assert_eq!(format_term("((a + b))"), format_term("a + b")); assert_eq!(format_term("((a - b))"), format_term("a - b")); assert_eq!(format_term("(f(a, b))"), format_term("f(a, b)")); assert_eq!(format_term("((f(a, b)))"), format_term("f(a, b)")); assert_eq!(format_term("f((a), (b))"), format_term("f(a, b)")); assert_eq!(format_term("f(|-a|)"), format_term("f(|-a|)")); assert_eq!(format_term("f((|-a|))"), format_term("f(|-a|)")); assert_eq!(format_term("f((-a))"), format_term("f(-a)")); assert_eq!(format_term("f(((-a)))"), format_term("f(-a)")); assert_eq!(format_term("f((a ** b))"), format_term("f(a ** b)")); assert_eq!(format_term("f((a * b))"), format_term("f(a * b)")); assert_eq!(format_term("f((a / b))"), format_term("f(a / b)")); assert_eq!(format_term("f((a % b))"), format_term("f(a % b)")); assert_eq!(format_term("f((a + b))"), format_term("f(a + b)")); assert_eq!(format_term("f((a - b))"), format_term("f(a - b)")); assert_eq!(format_term("f(((a ** b)))"), format_term("f(a ** b)")); assert_eq!(format_term("f(((a * b)))"), format_term("f(a * b)")); assert_eq!(format_term("f(((a / b)))"), format_term("f(a / b)")); assert_eq!(format_term("f(((a % b)))"), format_term("f(a % b)")); assert_eq!(format_term("f(((a + b)))"), format_term("f(a + b)")); assert_eq!(format_term("f(((a - b)))"), format_term("f(a - b)")); assert_eq!(format_term("(|a ** b|)"), format_term("|a ** b|")); assert_eq!(format_term("|(a ** b)|"), format_term("|a ** b|")); assert_eq!(format_term("(|(a ** b)|)"), format_term("|a ** b|")); assert_eq!(format_term("(|a * b|)"), format_term("|a * b|")); assert_eq!(format_term("|(a * b)|"), format_term("|a * b|")); assert_eq!(format_term("(|(a * b)|)"), format_term("|a * b|")); assert_eq!(format_term("(|a / b|)"), format_term("|a / b|")); assert_eq!(format_term("|(a / b)|"), format_term("|a / b|")); assert_eq!(format_term("(|(a / b)|)"), format_term("|a / b|")); assert_eq!(format_term("(|a % b|)"), format_term("|a % b|")); assert_eq!(format_term("|(a % b)|"), format_term("|a % b|")); assert_eq!(format_term("(|(a % b)|)"), format_term("|a % b|")); } #[test] fn parse_term_boolean() { assert_eq!(term("true"), Term::true_()); assert_eq!(term("false"), Term::false_()); } #[test] fn parse_term_integer() { assert_eq!(term("0"), Term::integer(0)); assert_eq!(term("10000"), Term::integer(10000)); assert_eq!(term("+10000"), Term::integer(10000)); assert_eq!(term("-10000"), Term::integer(-10000)); } #[test] fn parse_term_special_integer() { assert_eq!(term("#inf"), Term::infimum()); assert_eq!(term("#sup"), Term::supremum()); } #[test] fn parse_term_string() { assert_eq!(term("\"test 123\""), Term::string("test 123".to_string())); assert_eq!(term("\"123 test\""), Term::string("123 test".to_string())); assert_eq!(term("\" test 123 \""), Term::string(" test 123 ".to_string())); assert_eq!(term("\"test\\n123\""), Term::string("test\n123".to_string())); assert_eq!(term("\"test\\\"123\""), Term::string("test\"123".to_string())); assert_eq!(term("\"test\\\"123\\\"\""), Term::string("test\"123\"".to_string())); assert_eq!(term("\"\\\"test 123\\\"\""), Term::string("\"test 123\"".to_string())); assert_eq!(term("\"test\\\\123\""), Term::string("test\\123".to_string())); assert_eq!(term("\"test\\\\123\\\\\""), Term::string("test\\123\\".to_string())); assert_eq!(term("\"\\\\test 123\\\\\""), Term::string("\\test 123\\".to_string())); assert_eq!(term("\"test\\n123\""), Term::string("test\n123".to_string())); assert_eq!(term("\"test\\n123\\n\""), Term::string("test\n123\n".to_string())); assert_eq!(term("\"\\ntest 123\\n\""), Term::string("\ntest 123\n".to_string())); assert_eq!(term("\"test\\t123\""), Term::string("test\t123".to_string())); assert_eq!(term("\"test\\t123\\t\""), Term::string("test\t123\t".to_string())); assert_eq!(term("\"\\ttest 123\\t\""), Term::string("\ttest 123\t".to_string())); assert_eq!(term("\"πŸ™‚test 123\""), Term::string("πŸ™‚test 123".to_string())); assert_eq!(term("\"test πŸ™‚ 123\""), Term::string("test πŸ™‚ 123".to_string())); assert_eq!(term("\"test 123πŸ™‚\""), Term::string("test 123πŸ™‚".to_string())); } #[test] fn parse_term_function() { let term_as_function = |i| match term(i) { Term::Function(function) => function, _ => panic!("expected function"), }; assert_eq!(term_as_function("s").declaration.name, "s"); assert_eq!(term_as_function("s").declaration.arity, 0); assert_eq!(term_as_function("s()").declaration.name, "s"); assert_eq!(term_as_function("s").declaration.arity, 0); assert_eq!(term_as_function("s(1, 2, 3)").declaration.name, "s"); assert_eq!(term_as_function("s(1, 2, 3)").declaration.arity, 3); assert_eq!(*term_as_function("s(1, 2, 3)").arguments.remove(0), Term::integer(1)); assert_eq!(*term_as_function("s(1, 2, 3)").arguments.remove(2), Term::integer(3)); } #[test] fn parse_term_variable() { let term_as_variable = |i| match term(i) { Term::Variable(variable) => variable, _ => panic!("expected variable"), }; assert_eq!(term_as_variable("X").declaration.name, "X"); assert_eq!(term_as_variable("Variable_123").declaration.name, "Variable_123"); } #[test] fn parse_term_unary() { assert_eq!(format_term("|a|"), "|a|"); assert_eq!(format_term("||a||"), "||a||"); assert_eq!(format_term("|a - b|"), "|a - b|"); assert_eq!(format_term("|a| - b"), "|a| - b"); assert_eq!(format_term("a - |b|"), "a - |b|"); assert_eq!(format_term("||a| - b|"), "||a| - b|"); assert_eq!(format_term("|a - |b||"), "|a - |b||"); assert_eq!(format_term("||a| - |b||"), "||a| - |b||"); assert_eq!(format_term("||a| - |b| - |c||"), "||a| - |b| - |c||"); assert_eq!(format_term("||a - b| - |c - d||"), "||a - b| - |c - d||"); assert_eq!(format_term("-a"), "-a"); assert_eq!(format_term("--a"), "a"); assert_eq!(format_term("---a"), "-a"); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(format_term("-|a + b|"), "-|a + b|"); assert_eq!(format_term("--|a + b|"), "|a + b|"); assert_eq!(format_term("---|a + b|"), "-|a + b|"); assert_eq!(term("5"), Term::integer(5)); assert_eq!(term("-5"), Term::integer(-5)); assert_eq!(term("--5"), Term::integer(5)); assert_eq!(term("---5"), Term::integer(-5)); assert_eq!(term("0"), Term::integer(0)); assert_eq!(term("-0"), Term::integer(0)); assert_eq!(term("--0"), Term::integer(0)); assert_eq!(term("---0"), Term::integer(0)); } #[test] fn parse_term_exponentiate() { assert_eq!(term("1 ** (2 ** (3 ** (4 ** 5)))"), term("1 ** 2 ** 3 ** 4 ** 5")); assert_eq!(format_term("1 ** 2 ** 3 ** 4 ** 5"), "1 ** 2 ** 3 ** 4 ** 5"); assert_eq!(term("1 ** (2 ** (3 ** (4 ** 5)))"), term("1 ** 2 ** 3 ** 4 ** 5")); // As exponentiation is right-associative, these parentheses cannot be omitted assert_ne!(term("(((1 ** 2) ** 3) ** 4) ** 5"), term("1 ** 2 ** 3 ** 4 ** 5")); assert_eq!(format_term("(((1 ** 2) ** 3) ** 4) ** 5"), "(((1 ** 2) ** 3) ** 4) ** 5"); } #[test] fn parse_term_multiplicative() { assert_eq!(format_term("(a * b) * (c * d)"), "a * b * c * d"); assert_eq!(format_term("(a * b) * (c / d)"), "a * b * c / d"); assert_eq!(format_term("(a * b) * (c % d)"), "a * b * (c % d)"); assert_eq!(format_term("(a / b) * (c * d)"), "a / b * c * d"); assert_eq!(format_term("(a / b) * (c / d)"), "a / b * c / d"); assert_eq!(format_term("(a / b) * (c % d)"), "a / b * (c % d)"); assert_eq!(format_term("(a % b) * (c * d)"), "a % b * c * d"); assert_eq!(format_term("(a % b) * (c / d)"), "a % b * c / d"); assert_eq!(format_term("(a % b) * (c % d)"), "a % b * (c % d)"); assert_eq!(format_term("(a * b) / (c * d)"), "a * b / (c * d)"); assert_eq!(format_term("(a * b) / (c / d)"), "a * b / (c / d)"); assert_eq!(format_term("(a * b) / (c % d)"), "a * b / (c % d)"); assert_eq!(format_term("(a / b) / (c * d)"), "a / b / (c * d)"); assert_eq!(format_term("(a / b) / (c / d)"), "a / b / (c / d)"); assert_eq!(format_term("(a / b) / (c % d)"), "a / b / (c % d)"); assert_eq!(format_term("(a % b) / (c * d)"), "a % b / (c * d)"); assert_eq!(format_term("(a % b) / (c / d)"), "a % b / (c / d)"); assert_eq!(format_term("(a % b) / (c % d)"), "a % b / (c % d)"); assert_eq!(format_term("(a * b) % (c * d)"), "a * b % (c * d)"); assert_eq!(format_term("(a * b) % (c / d)"), "a * b % (c / d)"); assert_eq!(format_term("(a * b) % (c % d)"), "a * b % (c % d)"); assert_eq!(format_term("(a / b) % (c * d)"), "a / b % (c * d)"); assert_eq!(format_term("(a / b) % (c / d)"), "a / b % (c / d)"); assert_eq!(format_term("(a / b) % (c % d)"), "a / b % (c % d)"); assert_eq!(format_term("(a % b) % (c * d)"), "a % b % (c * d)"); assert_eq!(format_term("(a % b) % (c / d)"), "a % b % (c / d)"); assert_eq!(format_term("(a % b) % (c % d)"), "a % b % (c % d)"); } #[test] fn parse_term_additive() { assert_eq!(format_term("(a + b) + (c + d)"), "a + b + c + d"); assert_eq!(format_term("(a + b) + (c - d)"), "a + b + c - d"); assert_eq!(format_term("(a - b) + (c + d)"), "a - b + c + d"); assert_eq!(format_term("(a - b) + (c - d)"), "a - b + c - d"); assert_eq!(format_term("(a + b) - (c + d)"), "a + b - (c + d)"); assert_eq!(format_term("(a + b) - (c - d)"), "a + b - (c - d)"); assert_eq!(format_term("(a - b) - (c + d)"), "a - b - (c + d)"); assert_eq!(format_term("(a - b) - (c - d)"), "a - b - (c - d)"); } #[test] fn parse_term_precedence() { assert_eq!(term("-a + b"), term("(-a) + b")); assert_eq!(term("-a + -b"), term("(-a) + (-b)")); assert_eq!(term("-a + -b"), term("-(a) + -(b)")); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(term("-a - b"), term("(-a) - b")); assert_eq!(term("-a - -b"), term("(-a) - (-b)")); assert_eq!(term("-a - -b"), term("-(a) - -(b)")); assert_eq!(term("-a * b"), term("(-a) * b")); assert_eq!(term("-a * -b"), term("(-a) * (-b)")); assert_eq!(term("-a * -b"), term("-(a) * -(b)")); assert_eq!(term("-a / b"), term("(-a) / b")); assert_eq!(term("-a / -b"), term("(-a) / (-b)")); assert_eq!(term("-a / -b"), term("-(a) / -(b)")); assert_eq!(term("-a % b"), term("(-a) % b")); assert_eq!(term("-a % -b"), term("(-a) % (-b)")); assert_eq!(term("-a % -b"), term("-(a) % -(b)")); assert_eq!(term("-a ** b"), term("(-a) ** b")); assert_eq!(term("-a ** -b"), term("(-a) ** (-b)")); assert_eq!(term("-a ** -b"), term("-(a) ** -(b)")); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(format_term("-(a + b)"), "-(a + b)"); assert_eq!(term("a + (b * c) + d"), term("(a + (b * c)) + d")); assert_eq!(term("a + (b / c) + d"), term("(a + (b / c)) + d")); assert_eq!(term("a + (b % c) + d"), term("(a + (b % c)) + d")); assert_eq!(term("a - (b * c) - d"), term("(a - (b * c)) - d")); assert_eq!(term("a - (b / c) - d"), term("(a - (b / c)) - d")); assert_eq!(term("a - (b % c) - d"), term("(a - (b % c)) - d")); assert_eq!(format_term("(a + b) * (c + d)"), "(a + b) * (c + d)"); assert_eq!(format_term("(a + b) / (c + d)"), "(a + b) / (c + d)"); assert_eq!(format_term("(a + b) % (c + d)"), "(a + b) % (c + d)"); assert_eq!(format_term("(a - b) * (c - d)"), "(a - b) * (c - d)"); assert_eq!(format_term("(a - b) / (c - d)"), "(a - b) / (c - d)"); assert_eq!(format_term("(a - b) % (c - d)"), "(a - b) % (c - d)"); assert_eq!(term("a ** b ** c + d ** e ** f"), term("(a ** b ** c) + (d ** e ** f)")); assert_eq!(term("a ** (b ** c + d) ** e ** f"), term("a ** ((b ** c + d) ** (e ** f))")); assert_eq!(term("a ** b ** (c + d) ** e ** f"), term("a ** (b ** ((c + d) ** (e ** f)))")); assert_eq!(term("a ** b ** (c + d ** e) ** f"), term("a ** (b ** ((c + d ** e) ** f))")); assert_eq!(term("a ** b ** c - d ** e ** f"), term("(a ** b ** c) - (d ** e ** f)")); assert_eq!(term("a ** (b ** c - d) ** e ** f"), term("a ** ((b ** c - d) ** (e ** f))")); assert_eq!(term("a ** b ** (c - d) ** e ** f"), term("a ** (b ** ((c - d) ** (e ** f)))")); assert_eq!(term("a ** b ** (c - d ** e) ** f"), term("a ** (b ** ((c - d ** e) ** f))")); assert_eq!(term("a ** b ** c * d ** e ** f"), term("(a ** b ** c) * (d ** e ** f)")); assert_eq!(term("a ** (b ** c * d) ** e ** f"), term("a ** ((b ** c * d) ** (e ** f))")); assert_eq!(term("a ** b ** (c * d) ** e ** f"), term("a ** (b ** ((c * d) ** (e ** f)))")); assert_eq!(term("a ** b ** (c * d ** e) ** f"), term("a ** (b ** ((c * d ** e) ** f))")); assert_eq!(term("a ** b ** c / d ** e ** f"), term("(a ** b ** c) / (d ** e ** f)")); assert_eq!(term("a ** (b ** c / d) ** e ** f"), term("a ** ((b ** c / d) ** (e ** f))")); assert_eq!(term("a ** b ** (c / d) ** e ** f"), term("a ** (b ** ((c / d) ** (e ** f)))")); assert_eq!(term("a ** b ** (c / d ** e) ** f"), term("a ** (b ** ((c / d ** e) ** f))")); assert_eq!(term("a ** b ** c % d ** e ** f"), term("(a ** b ** c) % (d ** e ** f)")); assert_eq!(term("a ** (b ** c % d) ** e ** f"), term("a ** ((b ** c % d) ** (e ** f))")); assert_eq!(term("a ** b ** (c % d) ** e ** f"), term("a ** (b ** ((c % d) ** (e ** f)))")); assert_eq!(term("a ** b ** (c % d ** e) ** f"), term("a ** (b ** ((c % d ** e) ** f))")); } #[test] fn parse_term_bounds() { let term = |i| crate::parse::term(i, &Declarations::new()).unwrap().0; assert_eq!(term("1 ** 2 ** 3, rest"), ", rest"); assert_eq!(term("1 * 2 * 3, rest"), ", rest"); assert_eq!(term("1 / 2 / 3, rest"), ", rest"); assert_eq!(term("1 % 2 % 3, rest"), ", rest"); assert_eq!(term("1 + 2 + 3, rest"), ", rest"); assert_eq!(term("1 - 2 - 3, rest"), ", rest"); assert_eq!(term("1, rest"), ", rest"); assert_eq!(term("-1, rest"), ", rest"); assert_eq!(term("--1, rest"), ", rest"); assert_eq!(term("|1|, rest"), ", rest"); assert_eq!(term("|1| + |-2|, rest"), ", rest"); assert_eq!(term("||-2||, rest"), ", rest"); assert_eq!(term("|-|-2||, rest"), ", rest"); assert_eq!(term("(1), rest"), ", rest"); assert_eq!(term("a, rest"), ", rest"); assert_eq!(term("1, rest"), ", rest"); assert_eq!(term("true, rest"), ", rest"); assert_eq!(term("false, rest"), ", rest"); assert_eq!(term("#inf, rest"), ", rest"); assert_eq!(term("#sup, rest"), ", rest"); assert_eq!(term("f(1, 2), rest"), ", rest"); assert_eq!(term("g(1 ** 2, 3 * 4, #inf), rest"), ", rest"); assert_eq!(term("\"test\", rest"), ", rest"); assert_eq!(term("X, rest"), ", rest"); assert_eq!(term("Variable, rest"), ", rest"); assert_eq!(term("f(\"test\", Variable), rest"), ", rest"); } #[test] fn parse_term_whitespace() { assert_eq!(format_term("(a+b)*(c+d)"), "(a + b) * (c + d)"); assert_eq!(format_term("( a + b ) * ( c + d )"), "(a + b) * (c + d)"); assert_eq!(format_term("( a + b ) * ( c + d )"), "(a + b) * (c + d)"); assert_eq!(format_term("(\ta\t+\tb\t)\t*\t(\tc\t+\td\t)"), "(a + b) * (c + d)"); assert_eq!(format_term("(\na\n+\nb\n)\n*\n(\nc\n+\nd\n)"), "(a + b) * (c + d)"); assert_eq!(format_term("( \t a \t + \t b \t ) \t * \t ( \t c \t + \t d \t )"), "(a + b) * (c + d)"); assert_eq!(format_term("( \n a \n + \n b \n ) \n * \n ( \n c \n + \n d \n )"), "(a + b) * (c + d)"); assert_eq!(format_term("f(\ta\t+\tb\t,\tc\t+\td\t)"), "f(a + b, c + d)"); assert_eq!(format_term("f(\na\n+\nb\n,\nc\n+\nd\n)"), "f(a + b, c + d)"); assert_eq!(format_term("f( \t a \t + \t b \t , \t c \t + \t d \t)"), "f(a + b, c + d)"); assert_eq!(format_term("f( \n a \n + \n b \n , \n c \n + \n d \n)"), "f(a + b, c + d)"); } #[test] fn parse_boolean() { assert_eq!(boolean("true"), Ok(("", Term::true_()))); assert_eq!(boolean("false"), Ok(("", Term::false_()))); assert_eq!(boolean("true false"), Ok((" false", Term::true_()))); assert_eq!(boolean("false true"), Ok((" true", Term::false_()))); assert_eq!(boolean("true,"), Ok((",", Term::true_()))); assert_eq!(boolean("false,"), Ok((",", Term::false_()))); assert!(boolean("truefalse").is_err()); assert!(boolean("falsetrue").is_err()); assert!(boolean("truea").is_err()); assert!(boolean("falsea").is_err()); assert!(boolean("a").is_err()); assert!(boolean("-").is_err()); assert!(boolean(" ").is_err()); } #[test] fn parse_integer() { assert_eq!(integer("0"), Ok(("", Term::integer(0)))); assert_eq!(integer("10000"), Ok(("", Term::integer(10000)))); assert_eq!(integer("+10000"), Ok(("", Term::integer(10000)))); assert_eq!(integer("-10000"), Ok(("", Term::integer(-10000)))); assert_eq!(integer("0 42"), Ok((" 42", Term::integer(0)))); assert_eq!(integer("10000 42"), Ok((" 42", Term::integer(10000)))); assert_eq!(integer("+10000 42"), Ok((" 42", Term::integer(10000)))); assert_eq!(integer("-10000 42"), Ok((" 42", Term::integer(-10000)))); assert_eq!(integer("10000,"), Ok((",", Term::integer(10000)))); assert_eq!(integer("+10000,"), Ok((",", Term::integer(10000)))); assert_eq!(integer("-10000,"), Ok((",", Term::integer(-10000)))); assert!(integer("10000a").is_err()); assert!(integer("+10000a").is_err()); assert!(integer("-10000a").is_err()); assert!(integer("1.5").is_err()); assert!(integer("a").is_err()); assert!(integer("-").is_err()); assert!(integer(" ").is_err()); } #[test] fn parse_special_integer() { assert_eq!(special_integer("#inf"), Ok(("", Term::infimum()))); assert_eq!(special_integer("#sup"), Ok(("", Term::supremum()))); assert_eq!(special_integer("#inf #sup"), Ok((" #sup", Term::infimum()))); assert_eq!(special_integer("#sup #inf"), Ok((" #inf", Term::supremum()))); assert_eq!(special_integer("#inf,"), Ok((",", Term::infimum()))); assert_eq!(special_integer("#sup,"), Ok((",", Term::supremum()))); assert!(special_integer("#inf0").is_err()); assert!(special_integer("#sup0").is_err()); assert!(special_integer("#infimum").is_err()); assert!(special_integer("#supremum").is_err()); assert!(special_integer("inf").is_err()); assert!(special_integer("sup").is_err()); assert!(special_integer("0").is_err()); assert!(special_integer("10000").is_err()); assert!(special_integer("-10000").is_err()); assert!(special_integer("-").is_err()); assert!(special_integer("+").is_err()); assert!(special_integer("a").is_err()); assert!(special_integer(" ").is_err()); } #[test] fn parse_string() { assert_eq!(string("\"test 123\""), Ok(("", Term::string("test 123".to_string())))); assert_eq!(string("\"123 test\""), Ok(("", Term::string("123 test".to_string())))); assert_eq!(string("\" test 123 \""), Ok(("", Term::string(" test 123 ".to_string())))); assert_eq!(string("\"test 123\" \"rest"), Ok((" \"rest", Term::string("test 123".to_string())))); assert_eq!(string("\"test 123\", \"rest"), Ok((", \"rest", Term::string("test 123".to_string())))); assert_eq!(string("\"test\n123\""), Ok(("", Term::string("test\n123".to_string())))); assert_eq!(string("\"test\\\"123\""), Ok(("", Term::string("test\"123".to_string())))); assert_eq!(string("\"test\\\"123\\\"\""), Ok(("", Term::string("test\"123\"".to_string())))); assert_eq!(string("\"\\\"test 123\\\"\""), Ok(("", Term::string("\"test 123\"".to_string())))); assert_eq!(string("\"test\\\\123\""), Ok(("", Term::string("test\\123".to_string())))); assert_eq!(string("\"test\\\\123\\\\\""), Ok(("", Term::string("test\\123\\".to_string())))); assert_eq!(string("\"\\\\test 123\\\\\""), Ok(("", Term::string("\\test 123\\".to_string())))); assert_eq!(string("\"test\\n123\""), Ok(("", Term::string("test\n123".to_string())))); assert_eq!(string("\"test\\n123\\n\""), Ok(("", Term::string("test\n123\n".to_string())))); assert_eq!(string("\"\\ntest 123\\n\""), Ok(("", Term::string("\ntest 123\n".to_string())))); assert_eq!(string("\"test\\t123\""), Ok(("", Term::string("test\t123".to_string())))); assert_eq!(string("\"test\\t123\\t\""), Ok(("", Term::string("test\t123\t".to_string())))); assert_eq!(string("\"\\ttest 123\\t\""), Ok(("", Term::string("\ttest 123\t".to_string())))); assert_eq!(string("\"test πŸ™‚ 123\""), Ok(("", Term::string("test πŸ™‚ 123".to_string())))); assert_eq!(string("\"πŸ™‚test 123\""), Ok(("", Term::string("πŸ™‚test 123".to_string())))); assert_eq!(string("\"test 123πŸ™‚\""), Ok(("", Term::string("test 123πŸ™‚".to_string())))); assert!(string("\"test 123\"a").is_err()); assert!(string("\"test\\i123\"").is_err()); assert!(string("\"test").is_err()); assert!(string("test").is_err()); assert!(string("-").is_err()); assert!(string(" ").is_err()); } #[test] fn parse_function() { assert_eq!(function("s", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok(("", "s".to_string()))); assert_eq!(function("s", &Declarations::new()) .map(|(i, x)| (i, x.declaration.arity)), Ok(("", 0))); assert_eq!(function("s ( )", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok(("", "s".to_string()))); assert_eq!(function("s ( )", &Declarations::new()) .map(|(i, x)| (i, x.declaration.arity)), Ok(("", 0))); assert_eq!(function("s ( 1 , 2 , 3 )", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok(("", "s".to_string()))); assert_eq!(function("s ( 1 , 2 , 3 )", &Declarations::new()) .map(|(i, x)| (i, x.declaration.arity)), Ok(("", 3))); assert_eq!(function("s ( 1 , 2 , 3 )", &Declarations::new()) .map(|(i, mut x)| (i, x.arguments.remove(0))), Ok(("", Box::new(Term::integer(1))))); assert_eq!(function("s ( 1 , 2 , 3 )", &Declarations::new()) .map(|(i, mut x)| (i, x.arguments.remove(2))), Ok(("", Box::new(Term::integer(3))))); assert_eq!(function("s ( ), rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok((", rest", "s".to_string()))); assert_eq!(function("s ( ), rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.arity)), Ok((", rest", 0))); assert_eq!(function("s ( 1 , 2 , 3 ), rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok((", rest", "s".to_string()))); assert_eq!(function("s ( 1 , 2 , 3 ), rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.arity)), Ok((", rest", 3))); } #[test] fn parse_variable_declaration() { assert_eq!(variable_declaration("X Rest") .map(|(i, x)| (i, x.name.clone())), Ok((" Rest", "X".to_string()))); assert_eq!(variable_declaration("X, Rest") .map(|(i, x)| (i, x.name.clone())), Ok((", Rest", "X".to_string()))); // Variable declarations parsed at different locations should not be considered equal assert_ne!(variable_declaration("X"), variable_declaration("X")); assert_eq!(variable_declaration("Variable_123 Rest") .map(|(i, x)| (i, x.name.clone())), Ok((" Rest", "Variable_123".to_string()))); assert!(variable_declaration("0 Rest").is_err()); assert!(variable_declaration("123_Asd Rest").is_err()); assert!(variable_declaration("x Rest").is_err()); assert!(variable_declaration("variable_123 Rest").is_err()); assert!(variable_declaration("_ Rest").is_err()); assert!(variable_declaration("_variable_123 Rest").is_err()); assert!(variable_declaration(" ").is_err()); } #[test] fn parse_variable() { assert_eq!(variable("X Rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok((" Rest", "X".to_string()))); assert_eq!(variable("X, Rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok((", Rest", "X".to_string()))); assert_eq!(variable("Variable_123 Rest", &Declarations::new()) .map(|(i, x)| (i, x.declaration.name.clone())), Ok((" Rest", "Variable_123".to_string()))); assert!(variable("0 Rest", &Declarations::new()).is_err()); assert!(variable("123_Asd Rest", &Declarations::new()).is_err()); assert!(variable("x Rest", &Declarations::new()).is_err()); assert!(variable("variable_123 Rest", &Declarations::new()).is_err()); assert!(variable("_ Rest", &Declarations::new()).is_err()); assert!(variable("_variable_123 Rest", &Declarations::new()).is_err()); assert!(variable(" ", &Declarations::new()).is_err()); let new_variable_declarations = |names: &[&str]| std::rc::Rc::new(names.iter() .map(|name| std::rc::Rc::new(VariableDeclaration::new(name.to_string()))) .collect()); let layer_1 = new_variable_declarations(&["A", "B", "X"]); let layer_2 = new_variable_declarations(&["C", "D", "X"]); let layer_3 = new_variable_declarations(&["E", "F", "Y"]); let layer_4 = new_variable_declarations(&["G", "H", "X"]); let variable_declaration_stack = VariableDeclarationStack::new(); let mut declarations = Declarations::new(); declarations.variable_declaration_stack = std::cell::RefCell::new(variable_declaration_stack); let (_, x1) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 1); let (_, x2) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 1); assert_eq!(x1.declaration, x2.declaration); let (_, y1) = variable("Y", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(x1.declaration, y1.declaration); assert_ne!(x2.declaration, y1.declaration); declarations.variable_declaration_stack.borrow_mut().push(layer_1); let (_, x3) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(x1.declaration, x3.declaration); let (_, x4) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(x3.declaration, x4.declaration); let (_, a1) = variable("A", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(x3.declaration, a1.declaration); let (_, y2) = variable("Y", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(y1.declaration, y2.declaration); declarations.variable_declaration_stack.borrow_mut().push(layer_2); let (_, x5) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(x1.declaration, x5.declaration); assert_ne!(x3.declaration, x5.declaration); let (_, x6) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(x5.declaration, x6.declaration); let (_, a2) = variable("A", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(a1.declaration, a2.declaration); declarations.variable_declaration_stack.borrow_mut().push(layer_3); let (_, x7) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(x5.declaration, x7.declaration); let (_, y3) = variable("Y", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(y2.declaration, y3.declaration); declarations.variable_declaration_stack.borrow_mut().push(layer_4); let (_, x8) = variable("X", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_ne!(x7.declaration, x8.declaration); let (_, y4) = variable("Y", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 2); assert_eq!(y3.declaration, y4.declaration); let _ = variable("I", &declarations).unwrap(); assert_eq!(declarations.variable_declaration_stack.borrow() .free_variable_declarations.len(), 3); } }