553 lines
17 KiB
Rust
553 lines
17 KiB
Rust
extern crate evalexpr;
|
|
|
|
use evalexpr::{*, error::*};
|
|
|
|
#[test]
|
|
fn test_unary_examples() {
|
|
assert_eq!(eval("3"), Ok(Value::Int(3)));
|
|
assert_eq!(eval("3.3"), Ok(Value::Float(3.3)));
|
|
assert_eq!(eval("true"), Ok(Value::Boolean(true)));
|
|
assert_eq!(eval("false"), Ok(Value::Boolean(false)));
|
|
assert_eq!(
|
|
eval("blub"),
|
|
Err(EvalexprError::VariableIdentifierNotFound(
|
|
"blub".to_string()
|
|
))
|
|
);
|
|
assert_eq!(eval("-3"), Ok(Value::Int(-3)));
|
|
assert_eq!(eval("-3.6"), Ok(Value::Float(-3.6)));
|
|
assert_eq!(eval("----3"), Ok(Value::Int(3)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_binary_examples() {
|
|
assert_eq!(eval("1+3"), Ok(Value::Int(4)));
|
|
assert_eq!(eval("3+1"), Ok(Value::Int(4)));
|
|
assert_eq!(eval("3-5"), Ok(Value::Int(-2)));
|
|
assert_eq!(eval("5-3"), Ok(Value::Int(2)));
|
|
assert_eq!(eval("5 / 4"), Ok(Value::Int(1)));
|
|
assert_eq!(eval("5 *3"), Ok(Value::Int(15)));
|
|
assert_eq!(eval("1.0+3"), Ok(Value::Float(4.0)));
|
|
assert_eq!(eval("3.0+1"), Ok(Value::Float(4.0)));
|
|
assert_eq!(eval("3-5.0"), Ok(Value::Float(-2.0)));
|
|
assert_eq!(eval("5-3.0"), Ok(Value::Float(2.0)));
|
|
assert_eq!(eval("5 / 4.0"), Ok(Value::Float(1.25)));
|
|
assert_eq!(eval("5.0 *3"), Ok(Value::Float(15.0)));
|
|
assert_eq!(eval("5.0 *-3"), Ok(Value::Float(-15.0)));
|
|
assert_eq!(eval("5.0 *- 3"), Ok(Value::Float(-15.0)));
|
|
assert_eq!(eval("5.0 * -3"), Ok(Value::Float(-15.0)));
|
|
assert_eq!(eval("5.0 * - 3"), Ok(Value::Float(-15.0)));
|
|
assert_eq!(eval("-5.0 *-3"), Ok(Value::Float(15.0)));
|
|
assert_eq!(eval("3+-1"), Ok(Value::Int(2)));
|
|
assert_eq!(eval("-3-5"), Ok(Value::Int(-8)));
|
|
assert_eq!(eval("-5--3"), Ok(Value::Int(-2)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_arithmetic_precedence_examples() {
|
|
assert_eq!(eval("1+3-2"), Ok(Value::Int(2)));
|
|
assert_eq!(eval("3+1*5"), Ok(Value::Int(8)));
|
|
assert_eq!(eval("2*3-5"), Ok(Value::Int(1)));
|
|
assert_eq!(eval("5-3/3"), Ok(Value::Int(4)));
|
|
assert_eq!(eval("5 / 4*2"), Ok(Value::Int(2)));
|
|
assert_eq!(eval("1-5 *3/15"), Ok(Value::Int(0)));
|
|
assert_eq!(eval("15/7/2.0"), Ok(Value::Float(1.0)));
|
|
assert_eq!(eval("15.0/7/2"), Ok(Value::Float(15.0 / 7.0 / 2.0)));
|
|
assert_eq!(eval("15.0/-7/2"), Ok(Value::Float(15.0 / -7.0 / 2.0)));
|
|
assert_eq!(eval("-15.0/7/2"), Ok(Value::Float(-15.0 / 7.0 / 2.0)));
|
|
assert_eq!(eval("-15.0/7/-2"), Ok(Value::Float(-15.0 / 7.0 / -2.0)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_braced_examples() {
|
|
assert_eq!(eval("(1)"), Ok(Value::Int(1)));
|
|
assert_eq!(eval("( 1.0 )"), Ok(Value::Float(1.0)));
|
|
assert_eq!(eval("( true)"), Ok(Value::Boolean(true)));
|
|
assert_eq!(eval("( -1 )"), Ok(Value::Int(-1)));
|
|
assert_eq!(eval("-(1)"), Ok(Value::Int(-1)));
|
|
assert_eq!(eval("-(1 + 3) * 7"), Ok(Value::Int(-28)));
|
|
assert_eq!(eval("(1 * 1) - 3"), Ok(Value::Int(-2)));
|
|
assert_eq!(eval("4 / (2 * 2)"), Ok(Value::Int(1)));
|
|
assert_eq!(eval("7/(7/(7/(7/(7/(7)))))"), Ok(Value::Int(1)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_mod_examples() {
|
|
assert_eq!(eval("1 % 4"), Ok(Value::Int(1)));
|
|
assert_eq!(eval("6 % 4"), Ok(Value::Int(2)));
|
|
assert_eq!(eval("1 % 4 + 2"), Ok(Value::Int(3)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_pow_examples() {
|
|
assert_eq!(eval("1 ^ 4"), Ok(Value::Float(1.0)));
|
|
assert_eq!(eval("6 ^ 4"), Ok(Value::Float(6.0f64.powf(4.0))));
|
|
assert_eq!(eval("1 ^ 4 + 2"), Ok(Value::Float(3.0)));
|
|
assert_eq!(eval("2 ^ (4 + 2)"), Ok(Value::Float(64.0)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_boolean_examples() {
|
|
assert_eq!(eval("true && false"), Ok(Value::Boolean(false)));
|
|
assert_eq!(
|
|
eval("true && false || true && true"),
|
|
Ok(Value::Boolean(true))
|
|
);
|
|
assert_eq!(eval("5 > 4 && 1 <= 1"), Ok(Value::Boolean(true)));
|
|
assert_eq!(eval("5.0 <= 4.9 || !(4 > 3.5)"), Ok(Value::Boolean(false)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_with_context() {
|
|
let mut context = HashMapContext::new();
|
|
context
|
|
.set_value("tr".into(), Value::Boolean(true))
|
|
.unwrap();
|
|
context
|
|
.set_value("fa".into(), Value::Boolean(false))
|
|
.unwrap();
|
|
context.set_value("five".into(), Value::Int(5)).unwrap();
|
|
context.set_value("six".into(), Value::Int(6)).unwrap();
|
|
context.set_value("half".into(), Value::Float(0.5)).unwrap();
|
|
context.set_value("zero".into(), Value::Int(0)).unwrap();
|
|
|
|
assert_eq!(eval_with_context("tr", &context), Ok(Value::Boolean(true)));
|
|
assert_eq!(eval_with_context("fa", &context), Ok(Value::Boolean(false)));
|
|
assert_eq!(
|
|
eval_with_context("tr && false", &context),
|
|
Ok(Value::Boolean(false))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("five + six", &context),
|
|
Ok(Value::Int(11))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("five * half", &context),
|
|
Ok(Value::Float(2.5))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("five < six && true", &context),
|
|
Ok(Value::Boolean(true))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_functions() {
|
|
let mut context = HashMapContext::new();
|
|
context
|
|
.set_function(
|
|
"sub2".to_string(),
|
|
Function::new(
|
|
Some(1),
|
|
Box::new(|arguments| {
|
|
if let Value::Int(int) = arguments[0] {
|
|
Ok(Value::Int(int - 2))
|
|
} else if let Value::Float(float) = arguments[0] {
|
|
Ok(Value::Float(float - 2.0))
|
|
} else {
|
|
Err(EvalexprError::expected_number(arguments[0].clone()))
|
|
}
|
|
}),
|
|
),
|
|
)
|
|
.unwrap();
|
|
context
|
|
.set_value("five".to_string(), Value::Int(5))
|
|
.unwrap();
|
|
|
|
assert_eq!(eval_with_context("sub2 5", &context), Ok(Value::Int(3)));
|
|
assert_eq!(eval_with_context("sub2(5)", &context), Ok(Value::Int(3)));
|
|
assert_eq!(eval_with_context("sub2 five", &context), Ok(Value::Int(3)));
|
|
assert_eq!(eval_with_context("sub2(five)", &context), Ok(Value::Int(3)));
|
|
assert_eq!(
|
|
eval_with_context("sub2(3) + five", &context),
|
|
Ok(Value::Int(6))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_n_ary_functions() {
|
|
let mut context = HashMapContext::new();
|
|
context
|
|
.set_function(
|
|
"sub2".into(),
|
|
Function::new(
|
|
Some(1),
|
|
Box::new(|arguments| {
|
|
if let Value::Int(int) = arguments[0] {
|
|
Ok(Value::Int(int - 2))
|
|
} else if let Value::Float(float) = arguments[0] {
|
|
Ok(Value::Float(float - 2.0))
|
|
} else {
|
|
Err(EvalexprError::expected_number(arguments[0].clone()))
|
|
}
|
|
}),
|
|
),
|
|
)
|
|
.unwrap();
|
|
context
|
|
.set_function(
|
|
"avg".into(),
|
|
Function::new(
|
|
Some(2),
|
|
Box::new(|arguments| {
|
|
expect_number(&arguments[0])?;
|
|
expect_number(&arguments[1])?;
|
|
|
|
if let (Value::Int(a), Value::Int(b)) = (&arguments[0], &arguments[1]) {
|
|
Ok(Value::Int((a + b) / 2))
|
|
} else {
|
|
Ok(Value::Float(
|
|
(arguments[0].as_float()? + arguments[1].as_float()?) / 2.0,
|
|
))
|
|
}
|
|
}),
|
|
),
|
|
)
|
|
.unwrap();
|
|
context
|
|
.set_function(
|
|
"muladd".into(),
|
|
Function::new(
|
|
Some(3),
|
|
Box::new(|arguments| {
|
|
expect_number(&arguments[0])?;
|
|
expect_number(&arguments[1])?;
|
|
expect_number(&arguments[2])?;
|
|
|
|
if let (Value::Int(a), Value::Int(b), Value::Int(c)) =
|
|
(&arguments[0], &arguments[1], &arguments[2])
|
|
{
|
|
Ok(Value::Int(a * b + c))
|
|
} else {
|
|
Ok(Value::Float(
|
|
arguments[0].as_float()? * arguments[1].as_float()?
|
|
+ arguments[2].as_float()?,
|
|
))
|
|
}
|
|
}),
|
|
),
|
|
)
|
|
.unwrap();
|
|
context
|
|
.set_function(
|
|
"count".into(),
|
|
Function::new(
|
|
None,
|
|
Box::new(|arguments| {
|
|
if arguments.len() == 1 {
|
|
if arguments[0] == Value::Empty {
|
|
Ok(Value::Int(0))
|
|
} else {
|
|
Ok(Value::Int(1))
|
|
}
|
|
} else {
|
|
Ok(Value::Int(arguments.len() as IntType))
|
|
}
|
|
}),
|
|
),
|
|
)
|
|
.unwrap();
|
|
context
|
|
.set_value("five".to_string(), Value::Int(5))
|
|
.unwrap();
|
|
|
|
assert_eq!(eval_with_context("avg(7, 5)", &context), Ok(Value::Int(6)));
|
|
assert_eq!(
|
|
eval_with_context("avg(sub2 5, 5)", &context),
|
|
Ok(Value::Int(4))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("sub2(avg(3, 6))", &context),
|
|
Ok(Value::Int(2))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("sub2 avg(3, 6)", &context),
|
|
Ok(Value::Int(2))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("muladd(3, 6, -4)", &context),
|
|
Ok(Value::Int(14))
|
|
);
|
|
assert_eq!(eval_with_context("count()", &context), Ok(Value::Int(0)));
|
|
assert_eq!(
|
|
eval_with_context("count(3, 5.5, 2)", &context),
|
|
Ok(Value::Int(3))
|
|
);
|
|
assert_eq!(eval_with_context("count 5", &context), Ok(Value::Int(1)));
|
|
|
|
assert_eq!(
|
|
eval_with_context("min(4.0, 3)", &context),
|
|
Ok(Value::Int(3))
|
|
);
|
|
assert_eq!(
|
|
eval_with_context("max(4.0, 3)", &context),
|
|
Ok(Value::Float(4.0))
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_errors() {
|
|
assert_eq!(
|
|
eval("-true"),
|
|
Err(EvalexprError::expected_number(Value::Boolean(true)))
|
|
);
|
|
assert_eq!(
|
|
eval("1-true"),
|
|
Err(EvalexprError::expected_number(Value::Boolean(true)))
|
|
);
|
|
assert_eq!(
|
|
eval("true-"),
|
|
Err(EvalexprError::WrongOperatorArgumentAmount {
|
|
actual: 1,
|
|
expected: 2
|
|
})
|
|
);
|
|
assert_eq!(eval("!(()true)"), Err(EvalexprError::AppendedToLeafNode));
|
|
}
|
|
|
|
#[test]
|
|
fn test_no_panic() {
|
|
assert!(eval(&format!(
|
|
"{} + {}",
|
|
IntType::max_value(),
|
|
IntType::max_value()
|
|
))
|
|
.is_err());
|
|
assert!(eval(&format!(
|
|
"-{} - {}",
|
|
IntType::max_value(),
|
|
IntType::max_value()
|
|
))
|
|
.is_err());
|
|
assert!(eval(&format!("-(-{} - 1)", IntType::max_value())).is_err());
|
|
assert!(eval(&format!(
|
|
"{} * {}",
|
|
IntType::max_value(),
|
|
IntType::max_value()
|
|
))
|
|
.is_err());
|
|
assert!(eval(&format!("{} / {}", IntType::max_value(), 0)).is_err());
|
|
assert!(eval(&format!("{} % {}", IntType::max_value(), 0)).is_err());
|
|
assert!(eval(&format!(
|
|
"{} ^ {}",
|
|
IntType::max_value(),
|
|
IntType::max_value()
|
|
))
|
|
.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn test_shortcut_functions() {
|
|
let mut context = HashMapContext::new();
|
|
context
|
|
.set_value("string".into(), Value::from("a string"))
|
|
.unwrap();
|
|
|
|
// assert_eq!(eval_string("???"));
|
|
assert_eq!(
|
|
eval_string_with_context("string", &context),
|
|
Ok("a string".to_string())
|
|
);
|
|
assert_eq!(eval_float("3.3"), Ok(3.3));
|
|
assert_eq!(eval_float_with_context("3.3", &context), Ok(3.3));
|
|
assert_eq!(eval_int("3"), Ok(3));
|
|
assert_eq!(eval_int_with_context("3", &context), Ok(3));
|
|
assert_eq!(eval_number("3"), Ok(3.0));
|
|
assert_eq!(eval_number_with_context("3", &context), Ok(3.0));
|
|
assert_eq!(eval_boolean("true"), Ok(true));
|
|
assert_eq!(eval_boolean_with_context("true", &context), Ok(true));
|
|
assert_eq!(eval_tuple("3,3"), Ok(vec![Value::Int(3), Value::Int(3)]));
|
|
assert_eq!(
|
|
eval_tuple_with_context("3,3", &context),
|
|
Ok(vec![Value::Int(3), Value::Int(3)])
|
|
);
|
|
assert_eq!(eval_empty(""), Ok(EMPTY_VALUE));
|
|
assert_eq!(eval_empty_with_context("", &context), Ok(EMPTY_VALUE));
|
|
|
|
// assert_eq!(build_operator_tree("???").unwrap().eval_string());
|
|
assert_eq!(
|
|
build_operator_tree("string")
|
|
.unwrap()
|
|
.eval_string_with_context(&context),
|
|
Ok("a string".to_string())
|
|
);
|
|
assert_eq!(build_operator_tree("3.3").unwrap().eval_float(), Ok(3.3));
|
|
assert_eq!(
|
|
build_operator_tree("3.3")
|
|
.unwrap()
|
|
.eval_float_with_context(&context),
|
|
Ok(3.3)
|
|
);
|
|
assert_eq!(build_operator_tree("3").unwrap().eval_int(), Ok(3));
|
|
assert_eq!(
|
|
build_operator_tree("3")
|
|
.unwrap()
|
|
.eval_int_with_context(&context),
|
|
Ok(3)
|
|
);
|
|
assert_eq!(build_operator_tree("3").unwrap().eval_number(), Ok(3.0));
|
|
assert_eq!(
|
|
build_operator_tree("3")
|
|
.unwrap()
|
|
.eval_number_with_context(&context),
|
|
Ok(3.0)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("true").unwrap().eval_boolean(),
|
|
Ok(true)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("true")
|
|
.unwrap()
|
|
.eval_boolean_with_context(&context),
|
|
Ok(true)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("3,3").unwrap().eval_tuple(),
|
|
Ok(vec![Value::Int(3), Value::Int(3)])
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("3,3")
|
|
.unwrap()
|
|
.eval_tuple_with_context(&context),
|
|
Ok(vec![Value::Int(3), Value::Int(3)])
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("")
|
|
.unwrap()
|
|
.eval_empty_with_context(&context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
|
|
assert_eq!(
|
|
eval_string_with_context_mut("string", &mut context),
|
|
Ok("a string".to_string())
|
|
);
|
|
assert_eq!(eval_float_with_context_mut("3.3", &mut context), Ok(3.3));
|
|
assert_eq!(eval_int_with_context_mut("3", &mut context), Ok(3));
|
|
assert_eq!(eval_number_with_context_mut("3", &mut context), Ok(3.0));
|
|
assert_eq!(
|
|
eval_boolean_with_context_mut("true", &mut context),
|
|
Ok(true)
|
|
);
|
|
assert_eq!(
|
|
eval_tuple_with_context_mut("3,3", &mut context),
|
|
Ok(vec![Value::Int(3), Value::Int(3)])
|
|
);
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
|
|
assert_eq!(
|
|
build_operator_tree("string")
|
|
.unwrap()
|
|
.eval_string_with_context_mut(&mut context),
|
|
Ok("a string".to_string())
|
|
);
|
|
;
|
|
assert_eq!(
|
|
build_operator_tree("3.3")
|
|
.unwrap()
|
|
.eval_float_with_context_mut(&mut context),
|
|
Ok(3.3)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("3")
|
|
.unwrap()
|
|
.eval_int_with_context_mut(&mut context),
|
|
Ok(3)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("3")
|
|
.unwrap()
|
|
.eval_number_with_context_mut(&mut context),
|
|
Ok(3.0)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("true")
|
|
.unwrap()
|
|
.eval_boolean_with_context_mut(&mut context),
|
|
Ok(true)
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("3,3")
|
|
.unwrap()
|
|
.eval_tuple_with_context_mut(&mut context),
|
|
Ok(vec![Value::Int(3), Value::Int(3)])
|
|
);
|
|
assert_eq!(
|
|
build_operator_tree("")
|
|
.unwrap()
|
|
.eval_empty_with_context_mut(&mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_whitespace() {
|
|
assert!(eval_boolean("2 < = 3").is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_assignment() {
|
|
let mut context = HashMapContext::new();
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("int = 3", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("float = 2.0", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("tuple = (1,1)", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("empty = ()", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("boolean = false", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
|
|
assert_eq!(eval_int_with_context("int", &context), Ok(3));
|
|
assert_eq!(eval_float_with_context("float", &context), Ok(2.0));
|
|
assert_eq!(
|
|
eval_tuple_with_context("tuple", &context),
|
|
Ok(vec![1.into(), 1.into()])
|
|
);
|
|
assert_eq!(eval_empty_with_context("empty", &context), Ok(EMPTY_VALUE));
|
|
assert_eq!(eval_boolean_with_context("boolean", &context), Ok(false));
|
|
|
|
assert_eq!(
|
|
eval_empty_with_context_mut("b = a = 5", &mut context),
|
|
Ok(EMPTY_VALUE)
|
|
);
|
|
assert_eq!(eval_empty_with_context("b", &context), Ok(EMPTY_VALUE));
|
|
}
|
|
|
|
#[test]
|
|
fn test_expression_chaining() {
|
|
let mut context = HashMapContext::new();
|
|
assert_eq!(
|
|
eval_int_with_context_mut("a = 5; a = a + 2; a", &mut context),
|
|
Ok(7)
|
|
);
|
|
}
|
|
|
|
#[cfg(feature = "serde")]
|
|
#[test]
|
|
fn test_serde() {
|
|
let strings = ["3", "4+4", "21^(2*2)--3>5||!true"];
|
|
|
|
for string in &strings {
|
|
let manual_tree = build_operator_tree(string).unwrap();
|
|
let serde_tree: Node = ron::de::from_str(&format!("\"{}\"", string)).unwrap();
|
|
assert_eq!(manual_tree.eval(), serde_tree.eval());
|
|
}
|
|
}
|