1
0
dust/dust-lang/tests/functions.rs

170 lines
5.8 KiB
Rust
Raw Normal View History

use dust_lang::*;
#[test]
fn function() {
let source = "fn(a: int, b: int) -> int { a + b }";
assert_eq!(
run(source),
2024-11-16 06:29:21 +00:00
Ok(Some(ConcreteValue::Function(Chunk::with_data(
2024-11-11 00:28:21 +00:00
None,
2024-11-16 10:16:51 +00:00
FunctionType {
type_parameters: None,
value_parameters: None,
return_type: Box::new(Type::Function(FunctionType {
type_parameters: None,
value_parameters: Some(vec![(0, Type::Integer), (1, Type::Integer)]),
return_type: Box::new(Type::Integer),
}))
},
2024-11-11 00:28:21 +00:00
vec![
(
Instruction::add(
Destination::Register(2),
Argument::Local(0),
Argument::Local(1)
),
Span(30, 31)
),
(Instruction::r#return(true), Span(35, 35)),
2024-11-11 00:28:21 +00:00
],
2024-11-16 06:29:21 +00:00
vec![ConcreteValue::string("a"), ConcreteValue::string("b"),],
2024-11-11 00:28:21 +00:00
vec![
2024-11-26 01:43:18 +00:00
Local::new(0, Type::Integer, false, Scope::default()),
Local::new(1, Type::Integer, false, Scope::default())
2024-11-11 00:28:21 +00:00
]
))))
);
}
#[test]
fn function_call() {
let source = "fn(a: int, b: int) -> int { a + b }(1, 2)";
assert_eq!(
2024-11-06 20:40:37 +00:00
compile(source),
Ok(Chunk::with_data(
None,
2024-11-16 10:16:51 +00:00
FunctionType {
type_parameters: None,
value_parameters: None,
return_type: Box::new(Type::Integer)
},
vec![
2024-11-28 02:13:35 +00:00
(
Instruction::load_constant(Destination::Register(0), 0, false),
2024-11-28 02:13:35 +00:00
Span(0, 36)
),
(
Instruction::load_constant(Destination::Register(1), 1, false),
2024-11-28 02:13:35 +00:00
Span(36, 37)
),
(
Instruction::load_constant(Destination::Register(2), 2, false),
2024-11-28 02:13:35 +00:00
Span(39, 40)
),
(
Instruction::call(Destination::Register(3), Argument::Constant(0), 2),
2024-11-28 02:13:35 +00:00
Span(35, 41)
),
(Instruction::r#return(true), Span(41, 41)),
],
vec![
2024-11-16 06:29:21 +00:00
ConcreteValue::Function(Chunk::with_data(
2024-11-11 00:28:21 +00:00
None,
2024-11-16 10:16:51 +00:00
FunctionType {
type_parameters: None,
value_parameters: Some(vec![(0, Type::Integer), (1, Type::Integer)]),
return_type: Box::new(Type::Integer)
},
2024-11-11 00:28:21 +00:00
vec![
(
Instruction::add(
Destination::Register(2),
Argument::Local(0),
Argument::Local(1)
),
Span(30, 31)
),
(Instruction::r#return(true), Span(35, 36)),
2024-11-11 00:28:21 +00:00
],
2024-11-16 06:29:21 +00:00
vec![ConcreteValue::string("a"), ConcreteValue::string("b"),],
2024-11-11 00:28:21 +00:00
vec![
2024-11-26 01:43:18 +00:00
Local::new(0, Type::Integer, false, Scope::default()),
Local::new(1, Type::Integer, false, Scope::default())
2024-11-11 00:28:21 +00:00
]
)),
2024-11-16 06:29:21 +00:00
ConcreteValue::Integer(1),
ConcreteValue::Integer(2)
],
vec![]
)),
);
assert_eq!(run(source), Ok(Some(ConcreteValue::Integer(3))));
}
#[test]
fn function_declaration() {
let source = "fn add (a: int, b: int) -> int { a + b }";
assert_eq!(
2024-11-06 20:40:37 +00:00
compile(source),
Ok(Chunk::with_data(
None,
2024-11-16 10:16:51 +00:00
FunctionType {
type_parameters: None,
value_parameters: None,
return_type: Box::new(Type::None)
},
vec![
2024-11-28 02:13:35 +00:00
(
Instruction::load_constant(Destination::Register(0), 0, false),
2024-11-28 02:13:35 +00:00
Span(0, 40)
),
(Instruction::define_local(0, 0, false), Span(3, 6)),
(Instruction::r#return(false), Span(40, 40))
],
vec![
2024-11-16 06:29:21 +00:00
ConcreteValue::Function(Chunk::with_data(
Some("add".to_string()),
2024-11-16 10:16:51 +00:00
FunctionType {
type_parameters: None,
value_parameters: Some(vec![(0, Type::Integer), (1, Type::Integer)]),
return_type: Box::new(Type::Integer)
},
2024-11-11 00:28:21 +00:00
vec![
(
Instruction::add(
Destination::Register(2),
Argument::Local(0),
Argument::Local(1)
),
Span(35, 36)
),
(Instruction::r#return(true), Span(40, 40)),
2024-11-11 00:28:21 +00:00
],
2024-11-16 06:29:21 +00:00
vec![ConcreteValue::string("a"), ConcreteValue::string("b")],
2024-11-11 00:28:21 +00:00
vec![
2024-11-26 01:43:18 +00:00
Local::new(0, Type::Integer, false, Scope::default()),
Local::new(1, Type::Integer, false, Scope::default())
2024-11-11 00:28:21 +00:00
]
2024-11-16 10:16:51 +00:00
)),
ConcreteValue::string("add"),
],
vec![Local::new(
2024-11-16 10:16:51 +00:00
1,
Type::Function(FunctionType {
type_parameters: None,
value_parameters: Some(vec![(0, Type::Integer), (1, Type::Integer)]),
return_type: Box::new(Type::Integer),
}),
false,
Scope::default(),
2024-11-18 01:32:53 +00:00
),],
)),
);
assert_eq!(run(source), Ok(None));
}