Fix warnings, add destructuring let1
This commit is contained in:
@@ -49,13 +49,13 @@ impl Form {
|
|||||||
}
|
}
|
||||||
pub fn car(&self) -> Option<Rc<Form>> {
|
pub fn car(&self) -> Option<Rc<Form>> {
|
||||||
match self {
|
match self {
|
||||||
Form::Pair(car, cdr) => Some(Rc::clone(car)),
|
Form::Pair(car, _cdr) => Some(Rc::clone(car)),
|
||||||
_ => None,
|
_ => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn cdr(&self) -> Option<Rc<Form>> {
|
pub fn cdr(&self) -> Option<Rc<Form>> {
|
||||||
match self {
|
match self {
|
||||||
Form::Pair(car, cdr) => Some(Rc::clone(cdr)),
|
Form::Pair(_car, cdr) => Some(Rc::clone(cdr)),
|
||||||
_ => None,
|
_ => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
515
kr/src/main.rs
515
kr/src/main.rs
@@ -6,248 +6,6 @@ use std::rc::Rc;
|
|||||||
mod ast;
|
mod ast;
|
||||||
use crate::ast::{Form,PossibleTailCall};
|
use crate::ast::{Form,PossibleTailCall};
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn parse_test() {
|
|
||||||
let g = grammar::TermParser::new();
|
|
||||||
for test in [
|
|
||||||
"22", "(22)", "(((22)))",
|
|
||||||
"(22 )", "()", "( )", "( 44)", "(44 )",
|
|
||||||
"(22 44 (1) 33 (4 5 (6) 6))", "hello",
|
|
||||||
"-", "+", "(+ 1 ;hi
|
|
||||||
3)", "'13", "hello-world", "_",
|
|
||||||
] {
|
|
||||||
assert!(g.parse(test).is_ok());
|
|
||||||
}
|
|
||||||
assert!(g.parse("((22)").is_err());
|
|
||||||
|
|
||||||
let e = root_env();
|
|
||||||
|
|
||||||
fn eval_test<T: Into<Form>>(gram: &grammar::TermParser, e: &Rc<Form>, code: &str, expected: T) {
|
|
||||||
assert_eq!(*eval(Rc::clone(e), Rc::new(gram.parse(code).unwrap())), expected.into());
|
|
||||||
}
|
|
||||||
|
|
||||||
eval_test(&g, &e, "(+ 2 (car (cons 4 '(1 2))))", 6);
|
|
||||||
eval_test(&g, &e, "(= 17 ((vau d p (+ (eval (car p) d) 13)) (+ 1 3)))", true);
|
|
||||||
eval_test(&g, &e, "(if (= 2 2) (+ 1 2) (+ 3 4))", 3);
|
|
||||||
eval_test(&g, &e, "(quote a)", "a");
|
|
||||||
eval_test(&g, &e, "'a", "a");
|
|
||||||
eval_test(&g, &e, "'(1 . a)", (1, "a"));
|
|
||||||
eval_test(&g, &e, "'(1 a)", (1, ("a", Form::Nil)));
|
|
||||||
eval_test(&g, &e, "true", true);
|
|
||||||
eval_test(&g, &e, "false", false);
|
|
||||||
eval_test(&g, &e, "nil", Form::Nil);
|
|
||||||
|
|
||||||
eval_test(&g, &e, "(+ 1 2)", 3);
|
|
||||||
eval_test(&g, &e, "(- 1 2)", -1);
|
|
||||||
eval_test(&g, &e, "(* 1 2)", 2);
|
|
||||||
eval_test(&g, &e, "(/ 4 2)", 2);
|
|
||||||
eval_test(&g, &e, "(% 3 2)", 1);
|
|
||||||
eval_test(&g, &e, "(& 3 2)", 2);
|
|
||||||
eval_test(&g, &e, "(| 2 1)", 3);
|
|
||||||
eval_test(&g, &e, "(^ 2 1)", 3);
|
|
||||||
eval_test(&g, &e, "(^ 3 1)", 2);
|
|
||||||
|
|
||||||
eval_test(&g, &e, "(comb? +)", true);
|
|
||||||
eval_test(&g, &e, "(comb? (vau d p 1))", true);
|
|
||||||
eval_test(&g, &e, "(comb? 1)", false);
|
|
||||||
eval_test(&g, &e, "(pair? '(a))", true);
|
|
||||||
//eval_test(&g, &e, "(pair? '())", true);
|
|
||||||
eval_test(&g, &e, "(nil? nil)", true);
|
|
||||||
eval_test(&g, &e, "(nil? 1)", false);
|
|
||||||
eval_test(&g, &e, "(pair? 1)", false);
|
|
||||||
eval_test(&g, &e, "(symbol? 'a)", true);
|
|
||||||
eval_test(&g, &e, "(symbol? 1)", false);
|
|
||||||
eval_test(&g, &e, "(int? 1)", true);
|
|
||||||
eval_test(&g, &e, "(int? true)", false);
|
|
||||||
eval_test(&g, &e, "(bool? true)", true);
|
|
||||||
eval_test(&g, &e, "(bool? 1)", false);
|
|
||||||
|
|
||||||
eval_test(&g, &e, "!(bool?) 1", false);
|
|
||||||
eval_test(&g, &e, "!(bool?) true", true);
|
|
||||||
|
|
||||||
eval_test(&g, &e, "((vau root_env _ (eval 'a (cons (cons 'a 2) root_env))))", 2);
|
|
||||||
let LET = "!((vau root_env p (eval (car p)
|
|
||||||
(cons (cons 'let1
|
|
||||||
(vau de p (eval (car (cdr (cdr p))) (cons (cons (car p) (eval (car (cdr p)) de)) de)))
|
|
||||||
) root_env))))";
|
|
||||||
|
|
||||||
eval_test(&g, &e, &format!("{} (let1 x 10 (+ x 7))", LET), 17);
|
|
||||||
let FIB = "
|
|
||||||
!(let1 fib (vau de p
|
|
||||||
!(let1 self (eval (car p) de))
|
|
||||||
!(let1 n (eval (car (cdr p)) de))
|
|
||||||
!(if (= 0 n) 0)
|
|
||||||
!(if (= 1 n) 1)
|
|
||||||
(+ (self self (- n 1)) (self self (- n 2)))
|
|
||||||
))";
|
|
||||||
eval_test(&g, &e, &format!("{} {} (fib fib 6)", LET, FIB), 8);
|
|
||||||
|
|
||||||
let FACT = "
|
|
||||||
!(let1 fact (vau de p
|
|
||||||
!(let1 self (eval (car p) de))
|
|
||||||
!(let1 n (eval (car (cdr p)) de))
|
|
||||||
!(if (= 0 n) 1)
|
|
||||||
(* n (self self (- n 1)))
|
|
||||||
))";
|
|
||||||
eval_test(&g, &e, &format!("{} {} (fact fact 6)", LET, FACT), 720);
|
|
||||||
let VAPPLY = format!("
|
|
||||||
{}
|
|
||||||
!(let1 vapply (vau de p
|
|
||||||
!(let1 f (eval (car p) de))
|
|
||||||
!(let1 ip (eval (car (cdr p)) de))
|
|
||||||
!(let1 nde (eval (car (cdr (cdr p))) de))
|
|
||||||
(eval (cons f ip) nde)
|
|
||||||
))", LET);
|
|
||||||
|
|
||||||
// need the vapply to keep env in check because otherwise the env keeps growing
|
|
||||||
// and the Rc::drop will overflow the stack lol
|
|
||||||
let BADID = format!("
|
|
||||||
{}
|
|
||||||
!(let1 badid (vau de p
|
|
||||||
!(let1 inner (vau ide ip
|
|
||||||
!(let1 self (car ip))
|
|
||||||
!(let1 n (car (cdr ip)))
|
|
||||||
!(let1 acc (car (cdr (cdr ip))))
|
|
||||||
!(if (= 0 n) acc)
|
|
||||||
(vapply self (cons self (cons (- n 1) (cons (+ acc 1) nil))) de)
|
|
||||||
))
|
|
||||||
(vapply inner (cons inner (cons (eval (car p) de) (cons 0 nil))) de)
|
|
||||||
))", VAPPLY);
|
|
||||||
// Won't work unless tail calls work
|
|
||||||
eval_test(&g, &e, &format!("{} (badid 1000)", BADID), 1000);
|
|
||||||
|
|
||||||
// Maybe define in terms of a right fold?
|
|
||||||
let VMAP = format!("
|
|
||||||
{}
|
|
||||||
!(let1 vmap (vau de p
|
|
||||||
!(let1 vmap_inner (vau ide ip
|
|
||||||
!(let1 self (car ip))
|
|
||||||
!(let1 f (car (cdr ip)))
|
|
||||||
!(let1 l (car (cdr (cdr ip))))
|
|
||||||
!(if (= nil l) l)
|
|
||||||
(cons (vapply f (cons (car l) nil) de) (vapply self (cons self (cons f (cons (cdr l) nil))) de))
|
|
||||||
))
|
|
||||||
(vapply vmap_inner (cons vmap_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
|
||||||
))", VAPPLY);
|
|
||||||
eval_test(&g, &e, &format!("{} (vmap (vau de p (+ 1 (car p))) '(1 2 3))", VMAP), (2, (3, (4, Form::Nil))));
|
|
||||||
|
|
||||||
// Make sure (wrap (vau ...)) and internal style are optimized the same
|
|
||||||
let WRAP = format!("
|
|
||||||
{}
|
|
||||||
!(let1 wrap (vau de p
|
|
||||||
!(let1 f (eval (car p) de))
|
|
||||||
(vau ide p (vapply f (vmap (vau _ xp (eval (car xp) ide)) p) ide))
|
|
||||||
))", VMAP);
|
|
||||||
eval_test(&g, &e, &format!("{} ((wrap (vau _ p (+ (car p) 1))) (+ 1 2))", WRAP), 4);
|
|
||||||
// TODO: unwrap
|
|
||||||
|
|
||||||
let VFOLDL = format!("
|
|
||||||
{}
|
|
||||||
!(let1 vfoldl (vau de p
|
|
||||||
!(let1 vfoldl_inner (vau ide ip
|
|
||||||
!(let1 self (car ip))
|
|
||||||
!(let1 f (car (cdr ip)))
|
|
||||||
!(let1 a (car (cdr (cdr ip))))
|
|
||||||
!(let1 l (car (cdr (cdr (cdr ip)))))
|
|
||||||
!(if (= nil l) a)
|
|
||||||
(vapply self (cons self (cons f (cons (vapply f (cons a (cons (car l) nil)) de) (cons (cdr l) nil)))) de)
|
|
||||||
))
|
|
||||||
(vapply vfoldl_inner (cons vfoldl_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) (cons (eval (car (cdr (cdr p))) de) nil)))) de)
|
|
||||||
))", WRAP);
|
|
||||||
eval_test(&g, &e, &format!("{} (vfoldl (vau de p (+ (car p) (car (cdr p)))) 0 '(1 2 3))", VFOLDL), 6);
|
|
||||||
|
|
||||||
let ZIPD = format!("
|
|
||||||
{}
|
|
||||||
!(let1 zipd (vau de p
|
|
||||||
!(let1 zipd_inner (vau ide ip
|
|
||||||
!(let1 self (car ip))
|
|
||||||
!(let1 a (car (cdr ip)))
|
|
||||||
!(let1 b (car (cdr (cdr ip))))
|
|
||||||
!(if (= nil a) a)
|
|
||||||
!(if (= nil b) b)
|
|
||||||
(cons (cons (car a) (car b)) (vapply self (cons self (cons (cdr a) (cons (cdr b) nil))) de))
|
|
||||||
))
|
|
||||||
(vapply zipd_inner (cons zipd_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
|
||||||
))", VFOLDL);
|
|
||||||
eval_test(&g, &e, &format!("{} (zipd '(1 2 3) '(4 5 6))", ZIPD), ((1,4), ((2,5), ((3,6), Form::Nil))));
|
|
||||||
|
|
||||||
let CONCAT = format!("
|
|
||||||
{}
|
|
||||||
!(let1 concat (vau de p
|
|
||||||
!(let1 concat_inner (vau ide ip
|
|
||||||
!(let1 self (car ip))
|
|
||||||
!(let1 a (car (cdr ip)))
|
|
||||||
!(let1 b (car (cdr (cdr ip))))
|
|
||||||
!(if (= nil a) b)
|
|
||||||
(cons (car a) (vapply self (cons self (cons (cdr a) (cons b nil))) de))
|
|
||||||
))
|
|
||||||
(vapply concat_inner (cons concat_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
|
||||||
))", ZIPD);
|
|
||||||
eval_test(&g, &e, &format!("{} (concat '(1 2 3) '(4 5 6))", CONCAT), (1, (2, (3, (4, (5, (6, Form::Nil)))))));
|
|
||||||
|
|
||||||
// Should do some error checking, I suppose?
|
|
||||||
// Also, extend for variadic (no zip, but back to fold I think)
|
|
||||||
let BVAU = format!("
|
|
||||||
{}
|
|
||||||
!(let1 bvau (vau se p
|
|
||||||
!(let1 match_params (wrap (vau 0 p
|
|
||||||
!(let1 self (car p))
|
|
||||||
!(let1 p_ls (car (cdr p)))
|
|
||||||
!(let1 dp (car (cdr (cdr p))))
|
|
||||||
!(let1 e (car (cdr (cdr (cdr p)))))
|
|
||||||
!(if (= nil p_ls) e)
|
|
||||||
!(if (symbol? p_ls) (cons (cons p_ls dp) e))
|
|
||||||
(self self (cdr p_ls) (cdr dp) (self self (car p_ls) (car dp) e))
|
|
||||||
)))
|
|
||||||
(if (= nil (cdr (cdr p)))
|
|
||||||
; No de case
|
|
||||||
!(let1 p_ls (car p))
|
|
||||||
!(let1 b_v (car (cdr p)))
|
|
||||||
(vau 0 dp
|
|
||||||
(eval b_v (match_params match_params p_ls dp se))
|
|
||||||
)
|
|
||||||
|
|
||||||
; de case
|
|
||||||
!(let1 de_s (car p))
|
|
||||||
!(let1 p_ls (car (cdr p)))
|
|
||||||
!(let1 b_v (car (cdr (cdr p))))
|
|
||||||
(vau dde dp
|
|
||||||
(eval b_v (match_params match_params p_ls dp (cons (cons de_s dde) se)))
|
|
||||||
)
|
|
||||||
)
|
|
||||||
))", CONCAT);
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau _ (a b c) (+ a (- b c))) 10 2 3)", BVAU), 9);
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau (a b c) (+ a (- b c))) 10 2 3)", BVAU), 9);
|
|
||||||
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2 3)", BVAU), (3, Form::Nil));
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2)", BVAU), Form::Nil);
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2 3 4 5)", BVAU), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau c c) 3 4 5)", BVAU), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau c c))", BVAU), Form::Nil);
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) c) (10 2) 3 4 5)", BVAU), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) a) (10 2) 3 4 5)", BVAU), 10);
|
|
||||||
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) b) (10 2) 3 4 5)", BVAU), 2);
|
|
||||||
|
|
||||||
eval_test(&g, &e, &format!("{} ((wrap (bvau _ (a b c) (+ a (- b c)))) (+ 10 1) (+ 2 2) (+ 5 3))", BVAU), 7);
|
|
||||||
eval_test(&g, &e, &format!("{} ((wrap (bvau (a b c) (+ a (- b c)))) (+ 10 1) (+ 2 2) (+ 5 3))", BVAU), 7);
|
|
||||||
|
|
||||||
let LAMBDA = format!("
|
|
||||||
{}
|
|
||||||
!(let1 lambda (vau de p
|
|
||||||
(wrap (vapply bvau p de))
|
|
||||||
))", BVAU);
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda (a b c) (+ a (- b c))) (+ 10 1) (+ 2 2) (+ 5 3))", LAMBDA), 7);
|
|
||||||
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2 3)", LAMBDA), (3, Form::Nil));
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2)", LAMBDA), Form::Nil);
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2 3 4 5)", LAMBDA), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda c c) 3 4 5)", LAMBDA), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda c c))", LAMBDA), Form::Nil);
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) c) '(10 2) 3 4 5)", LAMBDA), (3, (4, (5, Form::Nil))));
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) a) '(10 2) 3 4 5)", LAMBDA), 10);
|
|
||||||
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) b) '(10 2) 3 4 5)", LAMBDA), 2);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn eval(e: Rc<Form>, f: Rc<Form>) -> Rc<Form> {
|
fn eval(e: Rc<Form>, f: Rc<Form>) -> Rc<Form> {
|
||||||
let mut e = e;
|
let mut e = e;
|
||||||
let mut x = Option::Some(f);
|
let mut x = Option::Some(f);
|
||||||
@@ -267,7 +25,7 @@ fn eval(e: Rc<Form>, f: Rc<Form>) -> Rc<Form> {
|
|||||||
Form::Pair(ref c, ref p) => {
|
Form::Pair(ref c, ref p) => {
|
||||||
let comb = eval(Rc::clone(&e), Rc::clone(c));
|
let comb = eval(Rc::clone(&e), Rc::clone(c));
|
||||||
match *comb {
|
match *comb {
|
||||||
Form::PrimComb(ref n, ref f) => match f(e, Rc::clone(p)) {
|
Form::PrimComb(ref _n, ref f) => match f(e, Rc::clone(p)) {
|
||||||
PossibleTailCall::Result(r) => return r,
|
PossibleTailCall::Result(r) => return r,
|
||||||
PossibleTailCall::TailCall(ne, nx) => {
|
PossibleTailCall::TailCall(ne, nx) => {
|
||||||
e = ne;
|
e = ne;
|
||||||
@@ -355,7 +113,7 @@ fn root_env() -> Rc<Form> {
|
|||||||
("cdr", Rc::new(Form::PrimComb("cdr".to_owned(), |e, p| {
|
("cdr", Rc::new(Form::PrimComb("cdr".to_owned(), |e, p| {
|
||||||
PossibleTailCall::Result(eval(Rc::clone(&e), p.car().unwrap()).cdr().unwrap())
|
PossibleTailCall::Result(eval(Rc::clone(&e), p.car().unwrap()).cdr().unwrap())
|
||||||
}))),
|
}))),
|
||||||
("quote", Rc::new(Form::PrimComb("quote".to_owned(), |e, p| {
|
("quote", Rc::new(Form::PrimComb("quote".to_owned(), |_e, p| {
|
||||||
PossibleTailCall::Result(p.car().unwrap())
|
PossibleTailCall::Result(p.car().unwrap())
|
||||||
}))),
|
}))),
|
||||||
|
|
||||||
@@ -407,7 +165,7 @@ fn root_env() -> Rc<Form> {
|
|||||||
|
|
||||||
("comb?", Rc::new(Form::PrimComb("comb?".to_owned(), |e, p| {
|
("comb?", Rc::new(Form::PrimComb("comb?".to_owned(), |e, p| {
|
||||||
PossibleTailCall::Result(Rc::new(Form::Bool(match &*eval(e, p.car().unwrap()) {
|
PossibleTailCall::Result(Rc::new(Form::Bool(match &*eval(e, p.car().unwrap()) {
|
||||||
Form::PrimComb(n, f) => true,
|
Form::PrimComb(_n, _f) => true,
|
||||||
Form::DeriComb { .. } => true,
|
Form::DeriComb { .. } => true,
|
||||||
_ => false,
|
_ => false,
|
||||||
})))
|
})))
|
||||||
@@ -452,13 +210,272 @@ fn root_env() -> Rc<Form> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
//let input = "(+ 2 (car (cons 4 '(1 2))))";
|
|
||||||
let input = "(= 17 ((vau d p (+ (eval (car p) d) 13)) (+ 1 3)))";
|
let input = "(= 17 ((vau d p (+ (eval (car p) d) 13)) (+ 1 3)))";
|
||||||
//let input = "(if (= 2 2) (+ 1 2) (+ 3 4))";
|
|
||||||
//let input = "(quote a)";
|
|
||||||
//let input = "'a";
|
|
||||||
let parsed_input = grammar::TermParser::new().parse(input).unwrap();
|
let parsed_input = grammar::TermParser::new().parse(input).unwrap();
|
||||||
println!("Parsed input is {:?}", parsed_input);
|
println!("Parsed input is {:?}", parsed_input);
|
||||||
let result = eval(root_env(), Rc::new(parsed_input));
|
let result = eval(root_env(), Rc::new(parsed_input));
|
||||||
println!("Result is {:?}", result);
|
println!("Result is {:?}", result);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parse_test() {
|
||||||
|
let g = grammar::TermParser::new();
|
||||||
|
for test in [
|
||||||
|
"22", "(22)", "(((22)))",
|
||||||
|
"(22 )", "()", "( )", "( 44)", "(44 )",
|
||||||
|
"(22 44 (1) 33 (4 5 (6) 6))", "hello",
|
||||||
|
"-", "+", "(+ 1 ;hi
|
||||||
|
3)", "'13", "hello-world", "_",
|
||||||
|
] {
|
||||||
|
assert!(g.parse(test).is_ok());
|
||||||
|
}
|
||||||
|
assert!(g.parse("((22)").is_err());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn eval_test() {
|
||||||
|
let g = grammar::TermParser::new();
|
||||||
|
let e = root_env();
|
||||||
|
|
||||||
|
fn eval_test<T: Into<Form>>(gram: &grammar::TermParser, e: &Rc<Form>, code: &str, expected: T) {
|
||||||
|
assert_eq!(*eval(Rc::clone(e), Rc::new(gram.parse(code).unwrap())), expected.into());
|
||||||
|
}
|
||||||
|
|
||||||
|
eval_test(&g, &e, "(+ 2 (car (cons 4 '(1 2))))", 6);
|
||||||
|
eval_test(&g, &e, "(= 17 ((vau d p (+ (eval (car p) d) 13)) (+ 1 3)))", true);
|
||||||
|
eval_test(&g, &e, "(if (= 2 2) (+ 1 2) (+ 3 4))", 3);
|
||||||
|
eval_test(&g, &e, "(quote a)", "a");
|
||||||
|
eval_test(&g, &e, "'a", "a");
|
||||||
|
eval_test(&g, &e, "'(1 . a)", (1, "a"));
|
||||||
|
eval_test(&g, &e, "'(1 a)", (1, ("a", Form::Nil)));
|
||||||
|
eval_test(&g, &e, "true", true);
|
||||||
|
eval_test(&g, &e, "false", false);
|
||||||
|
eval_test(&g, &e, "nil", Form::Nil);
|
||||||
|
|
||||||
|
eval_test(&g, &e, "(+ 1 2)", 3);
|
||||||
|
eval_test(&g, &e, "(- 1 2)", -1);
|
||||||
|
eval_test(&g, &e, "(* 1 2)", 2);
|
||||||
|
eval_test(&g, &e, "(/ 4 2)", 2);
|
||||||
|
eval_test(&g, &e, "(% 3 2)", 1);
|
||||||
|
eval_test(&g, &e, "(& 3 2)", 2);
|
||||||
|
eval_test(&g, &e, "(| 2 1)", 3);
|
||||||
|
eval_test(&g, &e, "(^ 2 1)", 3);
|
||||||
|
eval_test(&g, &e, "(^ 3 1)", 2);
|
||||||
|
|
||||||
|
eval_test(&g, &e, "(comb? +)", true);
|
||||||
|
eval_test(&g, &e, "(comb? (vau d p 1))", true);
|
||||||
|
eval_test(&g, &e, "(comb? 1)", false);
|
||||||
|
eval_test(&g, &e, "(pair? '(a))", true);
|
||||||
|
//eval_test(&g, &e, "(pair? '())", true);
|
||||||
|
eval_test(&g, &e, "(nil? nil)", true);
|
||||||
|
eval_test(&g, &e, "(nil? 1)", false);
|
||||||
|
eval_test(&g, &e, "(pair? 1)", false);
|
||||||
|
eval_test(&g, &e, "(symbol? 'a)", true);
|
||||||
|
eval_test(&g, &e, "(symbol? 1)", false);
|
||||||
|
eval_test(&g, &e, "(int? 1)", true);
|
||||||
|
eval_test(&g, &e, "(int? true)", false);
|
||||||
|
eval_test(&g, &e, "(bool? true)", true);
|
||||||
|
eval_test(&g, &e, "(bool? 1)", false);
|
||||||
|
|
||||||
|
eval_test(&g, &e, "!(bool?) 1", false);
|
||||||
|
eval_test(&g, &e, "!(bool?) true", true);
|
||||||
|
|
||||||
|
eval_test(&g, &e, "((vau root_env _ (eval 'a (cons (cons 'a 2) root_env))))", 2);
|
||||||
|
let def_let = "!((vau root_env p (eval (car p)
|
||||||
|
(cons (cons 'let1
|
||||||
|
(vau de p (eval (car (cdr (cdr p))) (cons (cons (car p) (eval (car (cdr p)) de)) de)))
|
||||||
|
) root_env))))";
|
||||||
|
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 x 10 (+ x 7))", def_let), 17);
|
||||||
|
let def_fib = "
|
||||||
|
!(let1 fib (vau de p
|
||||||
|
!(let1 self (eval (car p) de))
|
||||||
|
!(let1 n (eval (car (cdr p)) de))
|
||||||
|
!(if (= 0 n) 0)
|
||||||
|
!(if (= 1 n) 1)
|
||||||
|
(+ (self self (- n 1)) (self self (- n 2)))
|
||||||
|
))";
|
||||||
|
eval_test(&g, &e, &format!("{} {} (fib fib 6)", def_let, def_fib), 8);
|
||||||
|
|
||||||
|
let def_fact = "
|
||||||
|
!(let1 fact (vau de p
|
||||||
|
!(let1 self (eval (car p) de))
|
||||||
|
!(let1 n (eval (car (cdr p)) de))
|
||||||
|
!(if (= 0 n) 1)
|
||||||
|
(* n (self self (- n 1)))
|
||||||
|
))";
|
||||||
|
eval_test(&g, &e, &format!("{} {} (fact fact 6)", def_let, def_fact), 720);
|
||||||
|
let def_vapply = format!("
|
||||||
|
{}
|
||||||
|
!(let1 vapply (vau de p
|
||||||
|
!(let1 f (eval (car p) de))
|
||||||
|
!(let1 ip (eval (car (cdr p)) de))
|
||||||
|
!(let1 nde (eval (car (cdr (cdr p))) de))
|
||||||
|
(eval (cons f ip) nde)
|
||||||
|
))", def_let);
|
||||||
|
|
||||||
|
// need the vapply to keep env in check because otherwise the env keeps growing
|
||||||
|
// and the Rc::drop will overflow the stack lol
|
||||||
|
let def_badid = format!("
|
||||||
|
{}
|
||||||
|
!(let1 badid (vau de p
|
||||||
|
!(let1 inner (vau ide ip
|
||||||
|
!(let1 self (car ip))
|
||||||
|
!(let1 n (car (cdr ip)))
|
||||||
|
!(let1 acc (car (cdr (cdr ip))))
|
||||||
|
!(if (= 0 n) acc)
|
||||||
|
(vapply self (cons self (cons (- n 1) (cons (+ acc 1) nil))) de)
|
||||||
|
))
|
||||||
|
(vapply inner (cons inner (cons (eval (car p) de) (cons 0 nil))) de)
|
||||||
|
))", def_vapply);
|
||||||
|
// Won't work unless tail calls work
|
||||||
|
eval_test(&g, &e, &format!("{} (badid 1000)", def_badid), 1000);
|
||||||
|
|
||||||
|
// Maybe define in terms of a right fold?
|
||||||
|
let def_vmap = format!("
|
||||||
|
{}
|
||||||
|
!(let1 vmap (vau de p
|
||||||
|
!(let1 vmap_inner (vau ide ip
|
||||||
|
!(let1 self (car ip))
|
||||||
|
!(let1 f (car (cdr ip)))
|
||||||
|
!(let1 l (car (cdr (cdr ip))))
|
||||||
|
!(if (= nil l) l)
|
||||||
|
(cons (vapply f (cons (car l) nil) de) (vapply self (cons self (cons f (cons (cdr l) nil))) de))
|
||||||
|
))
|
||||||
|
(vapply vmap_inner (cons vmap_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
||||||
|
))", def_vapply);
|
||||||
|
eval_test(&g, &e, &format!("{} (vmap (vau de p (+ 1 (car p))) '(1 2 3))", def_vmap), (2, (3, (4, Form::Nil))));
|
||||||
|
|
||||||
|
// Make sure (wrap (vau ...)) and internal style are optimized the same
|
||||||
|
let def_wrap = format!("
|
||||||
|
{}
|
||||||
|
!(let1 wrap (vau de p
|
||||||
|
!(let1 f (eval (car p) de))
|
||||||
|
(vau ide p (vapply f (vmap (vau _ xp (eval (car xp) ide)) p) ide))
|
||||||
|
))", def_vmap);
|
||||||
|
eval_test(&g, &e, &format!("{} ((wrap (vau _ p (+ (car p) 1))) (+ 1 2))", def_wrap), 4);
|
||||||
|
// TODO: unwrap
|
||||||
|
|
||||||
|
let def_vfoldl = format!("
|
||||||
|
{}
|
||||||
|
!(let1 vfoldl (vau de p
|
||||||
|
!(let1 vfoldl_inner (vau ide ip
|
||||||
|
!(let1 self (car ip))
|
||||||
|
!(let1 f (car (cdr ip)))
|
||||||
|
!(let1 a (car (cdr (cdr ip))))
|
||||||
|
!(let1 l (car (cdr (cdr (cdr ip)))))
|
||||||
|
!(if (= nil l) a)
|
||||||
|
(vapply self (cons self (cons f (cons (vapply f (cons a (cons (car l) nil)) de) (cons (cdr l) nil)))) de)
|
||||||
|
))
|
||||||
|
(vapply vfoldl_inner (cons vfoldl_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) (cons (eval (car (cdr (cdr p))) de) nil)))) de)
|
||||||
|
))", def_wrap);
|
||||||
|
eval_test(&g, &e, &format!("{} (vfoldl (vau de p (+ (car p) (car (cdr p)))) 0 '(1 2 3))", def_vfoldl), 6);
|
||||||
|
|
||||||
|
let def_zipd = format!("
|
||||||
|
{}
|
||||||
|
!(let1 zipd (vau de p
|
||||||
|
!(let1 zipd_inner (vau ide ip
|
||||||
|
!(let1 self (car ip))
|
||||||
|
!(let1 a (car (cdr ip)))
|
||||||
|
!(let1 b (car (cdr (cdr ip))))
|
||||||
|
!(if (= nil a) a)
|
||||||
|
!(if (= nil b) b)
|
||||||
|
(cons (cons (car a) (car b)) (vapply self (cons self (cons (cdr a) (cons (cdr b) nil))) de))
|
||||||
|
))
|
||||||
|
(vapply zipd_inner (cons zipd_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
||||||
|
))", def_vfoldl);
|
||||||
|
eval_test(&g, &e, &format!("{} (zipd '(1 2 3) '(4 5 6))", def_zipd), ((1,4), ((2,5), ((3,6), Form::Nil))));
|
||||||
|
|
||||||
|
let def_concat = format!("
|
||||||
|
{}
|
||||||
|
!(let1 concat (vau de p
|
||||||
|
!(let1 concat_inner (vau ide ip
|
||||||
|
!(let1 self (car ip))
|
||||||
|
!(let1 a (car (cdr ip)))
|
||||||
|
!(let1 b (car (cdr (cdr ip))))
|
||||||
|
!(if (= nil a) b)
|
||||||
|
(cons (car a) (vapply self (cons self (cons (cdr a) (cons b nil))) de))
|
||||||
|
))
|
||||||
|
(vapply concat_inner (cons concat_inner (cons (eval (car p) de) (cons (eval (car (cdr p)) de) nil))) de)
|
||||||
|
))", def_zipd);
|
||||||
|
eval_test(&g, &e, &format!("{} (concat '(1 2 3) '(4 5 6))", def_concat), (1, (2, (3, (4, (5, (6, Form::Nil)))))));
|
||||||
|
|
||||||
|
// Should do some error checking, I suppose?
|
||||||
|
// Also, extend for variadic (no zip, but back to fold I think)
|
||||||
|
let def_bvau = format!("
|
||||||
|
{}
|
||||||
|
!(let1 match_params (wrap (vau 0 p
|
||||||
|
!(let1 self (car p))
|
||||||
|
!(let1 p_ls (car (cdr p)))
|
||||||
|
!(let1 dp (car (cdr (cdr p))))
|
||||||
|
!(let1 e (car (cdr (cdr (cdr p)))))
|
||||||
|
!(if (= nil p_ls) e)
|
||||||
|
!(if (symbol? p_ls) (cons (cons p_ls dp) e))
|
||||||
|
(self self (cdr p_ls) (cdr dp) (self self (car p_ls) (car dp) e))
|
||||||
|
)))
|
||||||
|
!(let1 bvau (vau se p
|
||||||
|
(if (= nil (cdr (cdr p)))
|
||||||
|
; No de case
|
||||||
|
!(let1 p_ls (car p))
|
||||||
|
!(let1 b_v (car (cdr p)))
|
||||||
|
(vau 0 dp
|
||||||
|
(eval b_v (match_params match_params p_ls dp se))
|
||||||
|
)
|
||||||
|
|
||||||
|
; de case
|
||||||
|
!(let1 de_s (car p))
|
||||||
|
!(let1 p_ls (car (cdr p)))
|
||||||
|
!(let1 b_v (car (cdr (cdr p))))
|
||||||
|
(vau dde dp
|
||||||
|
(eval b_v (match_params match_params p_ls dp (cons (cons de_s dde) se)))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
))", def_concat);
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau _ (a b c) (+ a (- b c))) 10 2 3)", def_bvau), 9);
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau (a b c) (+ a (- b c))) 10 2 3)", def_bvau), 9);
|
||||||
|
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2 3)", def_bvau), (3, Form::Nil));
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2)", def_bvau), Form::Nil);
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau (a b . c) c) 10 2 3 4 5)", def_bvau), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau c c) 3 4 5)", def_bvau), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau c c))", def_bvau), Form::Nil);
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) c) (10 2) 3 4 5)", def_bvau), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) a) (10 2) 3 4 5)", def_bvau), 10);
|
||||||
|
eval_test(&g, &e, &format!("{} ((bvau ((a b) . c) b) (10 2) 3 4 5)", def_bvau), 2);
|
||||||
|
|
||||||
|
eval_test(&g, &e, &format!("{} ((wrap (bvau _ (a b c) (+ a (- b c)))) (+ 10 1) (+ 2 2) (+ 5 3))", def_bvau), 7);
|
||||||
|
eval_test(&g, &e, &format!("{} ((wrap (bvau (a b c) (+ a (- b c)))) (+ 10 1) (+ 2 2) (+ 5 3))", def_bvau), 7);
|
||||||
|
|
||||||
|
let def_lambda = format!("
|
||||||
|
{}
|
||||||
|
!(let1 lambda (vau de p
|
||||||
|
(wrap (vapply bvau p de))
|
||||||
|
))", def_bvau);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda (a b c) (+ a (- b c))) (+ 10 1) (+ 2 2) (+ 5 3))", def_lambda), 7);
|
||||||
|
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2 3)", def_lambda), (3, Form::Nil));
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2)", def_lambda), Form::Nil);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda (a b . c) c) 10 2 3 4 5)", def_lambda), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda c c) 3 4 5)", def_lambda), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda c c))", def_lambda), Form::Nil);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) c) '(10 2) 3 4 5)", def_lambda), (3, (4, (5, Form::Nil))));
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) a) '(10 2) 3 4 5)", def_lambda), 10);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda ((a b) . c) b) '(10 2) 3 4 5)", def_lambda), 2);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda ((a b . c) d) b) '(10 2 3 4) 3)", def_lambda), 2);
|
||||||
|
eval_test(&g, &e, &format!("{} ((lambda ((a b . c) d) c) '(10 2 3 4) 3)", def_lambda), (3, (4, Form::Nil)));
|
||||||
|
|
||||||
|
let def_let2 = format!("
|
||||||
|
{}
|
||||||
|
!(let1 let1 (bvau dp (s v b)
|
||||||
|
(eval b (match_params match_params s (eval v dp) dp))
|
||||||
|
))
|
||||||
|
", def_lambda);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 x (+ 10 1) (+ x 1))", def_let2), 12);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 x '(10 1) x)", def_let2), (10, (1, Form::Nil)));
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 (a b) '(10 1) a)", def_let2), 10);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 (a b) '(10 1) b)", def_let2), 1);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 (a b . c) '(10 1) c)", def_let2), Form::Nil);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 (a b . c) '(10 1 2 3) c)", def_let2), (2, (3, Form::Nil)));
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 ((a . b) . c) '((10 1) 2 3) a)", def_let2), 10);
|
||||||
|
eval_test(&g, &e, &format!("{} (let1 ((a . b) . c) '((10 1) 2 3) b)", def_let2), (1, Form::Nil));
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user