feat: empty? func for list, vec, hashmap

This commit is contained in:
Roman Godmaire 2024-05-04 17:21:46 -04:00
parent 92b32ff2df
commit 59dfe1e854
3 changed files with 21 additions and 0 deletions

13
src/env/core.rs vendored
View file

@ -129,6 +129,19 @@ pub(super) fn core() -> HashMap<String, Node> {
Ok(Node::Boolean(false))
}),
),
(
"empty?",
Node::NativeFunc(|_env, args| {
arg_count!(1, args.len());
match args[0].borrow() {
Node::List(list) => Ok(Node::Boolean(list.is_empty())),
Node::Vector(vec) => Ok(Node::Boolean(vec.is_empty())),
Node::Map(map) => Ok(Node::Boolean(map.is_empty())),
_ => Err(Error::ExpectedCollection)?,
}
}),
),
// Ordering
(
"eq?",

View file

@ -8,6 +8,8 @@ pub enum Error {
InvalidOperator,
#[error("expected symbol")]
ExpectedSymbol,
#[error("expected list, vector, or map")]
ExpectedCollection,
#[error("expected list")]
ExpectedList,
#[error("expected number")]

View file

@ -112,14 +112,20 @@ mod test {
#[case("(list 1 2 3)", "(1 2 3)")]
#[case("(list? (list))", "true")]
#[case("(list? 23)", "false")]
#[case("(empty? (list 1 2 3))", "false")]
#[case("(empty? (list))", "true")]
// Vectors
#[case("(vector 1 2 3)", "[1 2 3]")]
#[case("(vector? [])", "true")]
#[case("(vector? 23)", "false")]
#[case("(empty? [1 2 3])", "false")]
#[case("(empty? [])", "true")]
// Hashmaps
#[case("(hashmap :a 1)", "{:a: 1}")]
#[case("(hashmap? {})", "true")]
#[case("(hashmap? 23)", "false")]
#[case("(empty? {:a 1})", "false")]
#[case("(empty? {})", "true")]
// Environment manipulation
#[case("(define! asdf (+ 2 2)) (+ asdf 2)", "4\n6")]
#[case("(let* (a 2) (+ a 2))", "4")]