dust/tests/variables.rs
2024-03-08 19:05:17 -05:00

48 lines
1.1 KiB
Rust

use dust_lang::{
abstract_tree::{Expression, Identifier, Statement, Type},
error::{Error, TypeCheckError, ValidationError},
*,
};
#[test]
fn set_and_get_variable() {
assert_eq!(
interpret("foobar = true; foobar"),
Ok(Some(Value::boolean(true)))
);
}
#[test]
fn set_variable_with_type() {
assert_eq!(
interpret("foobar: bool = true; foobar"),
Ok(Some(Value::boolean(true)))
);
}
#[test]
fn set_variable_with_type_error() {
assert_eq!(
interpret("foobar: str = true"),
Err(vec![Error::Validation {
error: ValidationError::TypeCheck(TypeCheckError {
actual: Type::Boolean,
expected: Type::String
}),
span: (0..18).into()
}])
);
}
#[test]
fn function_variable() {
assert_eq!(
interpret("foobar = (x: int): int x; foobar"),
Ok(Some(Value::function(
vec![(Identifier::new("x"), Type::Integer)],
Type::Integer,
Statement::Expression(Expression::Identifier(Identifier::new("x")))
)))
);
}