1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#![allow(non_upper_case_globals)]
use chrysanthemum::ast::*;
use chrysanthemum::parser::*;
use chrysanthemum::simple::*;
use chrysanthemum::util::*;
// rust you KNOW these are &'static strs
const sanity_check: &'static str = "413: int";
const negate: &'static str = "(λx. if x then false else true): (bool -> bool)";
const basic_abstraction: &'static str = "(λx. x): (int -> int)";
const basic_application: &'static str = "((λx. x): (int -> int)) 413";
const correct_cond_abs: &'static str = "(λx. if x then 1 else 0): (bool -> int)";
const correct_cond: &'static str = "if false then 1: nat else 0: nat";
const not_inferrable: &'static str = "(λx. (λy. (λz. if x then y else z)))";
const incorrect_branches: &'static str = "if false: bool then true: bool else 2: int";
const incorrect_cond_abs: &'static str = "(λx. if x then true: bool else false: bool): (int -> bool)";
#[test]
fn test_parsing_succeeds() {
assert!(parse_lambda(sanity_check).is_ok());
assert!(parse_lambda(negate).is_ok());
assert!(parse_lambda(basic_abstraction).is_ok());
assert!(parse_lambda(basic_application).is_ok());
assert!(parse_lambda(correct_cond_abs).is_ok());
assert!(parse_lambda(correct_cond).is_ok());
assert!(parse_lambda(not_inferrable).is_ok());
assert!(parse_lambda(incorrect_branches).is_ok());
assert!(parse_lambda(incorrect_cond_abs).is_ok());
}
#[test]
fn test_inference() {
let context = Context::new();
assert_eq!(infer(&context, parse_lambda(sanity_check).unwrap()), Ok(Int));
assert_eq!(infer(&context, parse_lambda(negate).unwrap()), Ok(Func(Bool, Bool)));
assert_eq!(infer(&context, parse_lambda(basic_abstraction).unwrap()), Ok(Func(Int, Int)));
assert_eq!(infer(&context, parse_lambda(basic_application).unwrap()), Ok(Int));
assert_eq!(infer(&context, parse_lambda(correct_cond_abs).unwrap()), Ok(Func(Bool, Int)));
assert_eq!(infer(&context, parse_lambda(correct_cond).unwrap()), Ok(Nat));
assert!(infer(&context, parse_lambda(not_inferrable).unwrap()).is_err());
assert!(infer(&context, parse_lambda(incorrect_branches).unwrap()).is_err());
assert!(infer(&context, parse_lambda(incorrect_cond_abs).unwrap()).is_err());
}
#[test]
fn test_checking() {
let context = Context::new();
// uninteresting
assert!(check(&context, parse_lambda(sanity_check).unwrap(), &Int).is_ok());
assert!(check(&context, parse_lambda(negate).unwrap(), &Func(Bool, Bool)).is_ok());
assert!(check(&context, parse_lambda(basic_abstraction).unwrap(), &Func(Int, Int)).is_ok());
assert!(check(&context, parse_lambda(basic_application).unwrap(), &Int).is_ok());
assert!(check(&context, parse_lambda(correct_cond_abs).unwrap(), &Func(Bool, Int)).is_ok());
assert!(check(&context, parse_lambda(correct_cond).unwrap(), &Nat).is_ok());
assert!(check(&context, parse_lambda(incorrect_branches).unwrap(), &Unit).is_err());
assert!(check(&context, parse_lambda(incorrect_cond_abs).unwrap(), &Error).is_err());
// more fun
assert!(check(&context, parse_lambda(not_inferrable).unwrap(), &Func(Bool, Func(Int, Func(Int, Int)))).is_ok());
assert!(check(&context, parse_lambda(not_inferrable).unwrap(), &Func(Bool, Func(Nat, Func(Nat, Nat)))).is_ok());
assert!(check(&context, parse_lambda(not_inferrable).unwrap(), &Func(Bool, Func(Unit, Func(Unit, Unit)))).is_ok());
}
|
