mirror of
https://github.com/roc-lang/roc.git
synced 2025-09-27 22:09:09 +00:00
2581 lines
78 KiB
Rust
2581 lines
78 KiB
Rust
#[macro_use]
|
|
extern crate pretty_assertions;
|
|
|
|
#[macro_use]
|
|
extern crate indoc;
|
|
|
|
extern crate bumpalo;
|
|
extern crate roc_mono;
|
|
|
|
mod helpers;
|
|
|
|
// Test monomorphization
|
|
#[cfg(test)]
|
|
mod test_mono {
|
|
use roc_can::builtins::builtin_defs_map;
|
|
use roc_collections::all::MutMap;
|
|
use roc_module::symbol::Symbol;
|
|
use roc_mono::ir::Proc;
|
|
|
|
use roc_mono::ir::TopLevelFunctionLayout;
|
|
|
|
fn promote_expr_to_module(src: &str) -> String {
|
|
let mut buffer =
|
|
String::from("app \"test\" provides [ main ] to \"./platform\"\n\nmain =\n");
|
|
|
|
for line in src.lines() {
|
|
// indent the body!
|
|
buffer.push_str(" ");
|
|
buffer.push_str(line);
|
|
buffer.push('\n');
|
|
}
|
|
|
|
buffer
|
|
}
|
|
|
|
fn compiles_to_ir(src: &str, expected: &str) {
|
|
use bumpalo::Bump;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
let arena = &Bump::new();
|
|
|
|
// let stdlib = roc_builtins::unique::uniq_stdlib();
|
|
let stdlib = roc_builtins::std::standard_stdlib();
|
|
let filename = PathBuf::from("Test.roc");
|
|
let src_dir = Path::new("fake/test/path");
|
|
|
|
let module_src;
|
|
let temp;
|
|
if src.starts_with("app") {
|
|
// this is already a module
|
|
module_src = src;
|
|
} else {
|
|
// this is an expression, promote it to a module
|
|
temp = promote_expr_to_module(src);
|
|
module_src = &temp;
|
|
}
|
|
|
|
let exposed_types = MutMap::default();
|
|
|
|
let loaded = roc_load::file::load_and_monomorphize_from_str(
|
|
arena,
|
|
filename,
|
|
&module_src,
|
|
&stdlib,
|
|
src_dir,
|
|
exposed_types,
|
|
8,
|
|
builtin_defs_map,
|
|
);
|
|
|
|
let mut loaded = match loaded {
|
|
Ok(x) => x,
|
|
Err(roc_load::file::LoadingProblem::FormattedReport(report)) => {
|
|
println!("{}", report);
|
|
panic!();
|
|
}
|
|
Err(e) => panic!("{:?}", e),
|
|
};
|
|
|
|
use roc_load::file::MonomorphizedModule;
|
|
let MonomorphizedModule {
|
|
module_id: home,
|
|
procedures,
|
|
exposed_to_host,
|
|
..
|
|
} = loaded;
|
|
|
|
let can_problems = loaded.can_problems.remove(&home).unwrap_or_default();
|
|
let type_problems = loaded.type_problems.remove(&home).unwrap_or_default();
|
|
let mono_problems = loaded.mono_problems.remove(&home).unwrap_or_default();
|
|
|
|
if !can_problems.is_empty() {
|
|
println!("Ignoring {} canonicalization problems", can_problems.len());
|
|
}
|
|
|
|
assert_eq!(type_problems, Vec::new());
|
|
assert_eq!(mono_problems, Vec::new());
|
|
|
|
debug_assert_eq!(exposed_to_host.len(), 1);
|
|
|
|
let main_fn_symbol = exposed_to_host.keys().copied().next().unwrap();
|
|
|
|
verify_procedures(expected, procedures, main_fn_symbol);
|
|
}
|
|
|
|
#[cfg(debug_assertions)]
|
|
fn verify_procedures(
|
|
expected: &str,
|
|
procedures: MutMap<(Symbol, TopLevelFunctionLayout<'_>), Proc<'_>>,
|
|
main_fn_symbol: Symbol,
|
|
) {
|
|
let index = procedures
|
|
.keys()
|
|
.position(|(s, _)| *s == main_fn_symbol)
|
|
.unwrap();
|
|
|
|
let mut procs_string = procedures
|
|
.values()
|
|
.map(|proc| proc.to_pretty(200))
|
|
.collect::<Vec<_>>();
|
|
|
|
let main_fn = procs_string.swap_remove(index);
|
|
|
|
procs_string.sort();
|
|
procs_string.push(main_fn);
|
|
|
|
let result = procs_string.join("\n");
|
|
|
|
let the_same = result == expected;
|
|
|
|
if !the_same {
|
|
let expected_lines = expected.split('\n').collect::<Vec<&str>>();
|
|
let result_lines = result.split('\n').collect::<Vec<&str>>();
|
|
|
|
for line in &result_lines {
|
|
if !line.is_empty() {
|
|
println!(" {}", line);
|
|
} else {
|
|
println!();
|
|
}
|
|
}
|
|
|
|
assert_eq!(expected_lines, result_lines);
|
|
assert_eq!(0, 1);
|
|
}
|
|
}
|
|
|
|
// NOTE because the Show instance of module names is different in --release mode,
|
|
// these tests would all fail. In the future, when we do interesting optimizations,
|
|
// we'll likely want some tests for --release too.
|
|
#[cfg(not(debug_assertions))]
|
|
fn verify_procedures(
|
|
_expected: &str,
|
|
_procedures: MutMap<(Symbol, TopLevelFunctionLayout<'_>), Proc<'_>>,
|
|
_main_fn_symbol: Symbol,
|
|
) {
|
|
// Do nothing
|
|
}
|
|
|
|
#[test]
|
|
fn ir_int_literal() {
|
|
compiles_to_ir(
|
|
r#"
|
|
5
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_int_add() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x = [ 1,2 ]
|
|
5 + 4 + 3 + List.len x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure List.7 (#Attr.2):
|
|
let Test.6 = lowlevel ListLen #Attr.2;
|
|
ret Test.6;
|
|
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.5 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.5;
|
|
|
|
procedure Test.0 ():
|
|
let Test.11 = 1i64;
|
|
let Test.12 = 2i64;
|
|
let Test.1 = Array [Test.11, Test.12];
|
|
let Test.9 = 5i64;
|
|
let Test.10 = 4i64;
|
|
invoke Test.7 = CallByName Num.24 Test.9 Test.10 catch
|
|
dec Test.1;
|
|
unreachable;
|
|
let Test.8 = 3i64;
|
|
invoke Test.3 = CallByName Num.24 Test.7 Test.8 catch
|
|
dec Test.1;
|
|
unreachable;
|
|
let Test.4 = CallByName List.7 Test.1;
|
|
dec Test.1;
|
|
let Test.2 = CallByName Num.24 Test.3 Test.4;
|
|
ret Test.2;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_assignment() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x = 5
|
|
|
|
x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_when_maybe() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when Just 3 is
|
|
Just n -> n
|
|
Nothing -> 0
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.10 = 0i64;
|
|
let Test.9 = 3i64;
|
|
let Test.3 = Just Test.10 Test.9;
|
|
let Test.6 = 0i64;
|
|
let Test.7 = Index 0 Test.3;
|
|
let Test.8 = lowlevel Eq Test.6 Test.7;
|
|
if Test.8 then
|
|
let Test.2 = Index 1 Test.3;
|
|
ret Test.2;
|
|
else
|
|
let Test.5 = 0i64;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_when_these() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when These 1 2 is
|
|
This x -> x
|
|
That y -> y
|
|
These x _ -> x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.11 = 1i64;
|
|
let Test.9 = 1i64;
|
|
let Test.10 = 2i64;
|
|
let Test.5 = These Test.11 Test.9 Test.10;
|
|
switch Test.5:
|
|
case 2:
|
|
let Test.2 = Index 1 Test.5;
|
|
ret Test.2;
|
|
|
|
case 0:
|
|
let Test.3 = Index 1 Test.5;
|
|
ret Test.3;
|
|
|
|
default:
|
|
let Test.4 = Index 1 Test.5;
|
|
ret Test.4;
|
|
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_when_record() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when { x: 1, y: 3.14 } is
|
|
{ x } -> x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.4 = 1i64;
|
|
let Test.5 = 3.14f64;
|
|
let Test.2 = Struct {Test.4, Test.5};
|
|
let Test.1 = Index 0 Test.2;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_plus() {
|
|
compiles_to_ir(
|
|
r#"
|
|
1 + 2
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.4 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.4;
|
|
|
|
procedure Test.0 ():
|
|
let Test.2 = 1i64;
|
|
let Test.3 = 2i64;
|
|
let Test.1 = CallByName Num.24 Test.2 Test.3;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_round() {
|
|
compiles_to_ir(
|
|
r#"
|
|
Num.round 3.6
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.47 (#Attr.2):
|
|
let Test.3 = lowlevel NumRound #Attr.2;
|
|
ret Test.3;
|
|
|
|
procedure Test.0 ():
|
|
let Test.2 = 3.6f64;
|
|
let Test.1 = CallByName Num.47 Test.2;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_when_idiv() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when 1000 // 10 is
|
|
Ok val -> val
|
|
Err _ -> -1
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.42 (#Attr.2, #Attr.3):
|
|
let Test.17 = 0i64;
|
|
let Test.13 = lowlevel NotEq #Attr.3 Test.17;
|
|
if Test.13 then
|
|
let Test.16 = 1i64;
|
|
let Test.15 = lowlevel NumDivUnchecked #Attr.2 #Attr.3;
|
|
let Test.14 = Ok Test.16 Test.15;
|
|
ret Test.14;
|
|
else
|
|
let Test.12 = 0i64;
|
|
let Test.11 = Struct {};
|
|
let Test.10 = Err Test.12 Test.11;
|
|
ret Test.10;
|
|
|
|
procedure Test.0 ():
|
|
let Test.8 = 1000i64;
|
|
let Test.9 = 10i64;
|
|
let Test.2 = CallByName Num.42 Test.8 Test.9;
|
|
let Test.5 = 1i64;
|
|
let Test.6 = Index 0 Test.2;
|
|
let Test.7 = lowlevel Eq Test.5 Test.6;
|
|
if Test.7 then
|
|
let Test.1 = Index 1 Test.2;
|
|
ret Test.1;
|
|
else
|
|
let Test.4 = -1i64;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_two_defs() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x = 3
|
|
y = 4
|
|
|
|
x + y
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.4 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.4;
|
|
|
|
procedure Test.0 ():
|
|
let Test.1 = 3i64;
|
|
let Test.2 = 4i64;
|
|
let Test.3 = CallByName Num.24 Test.1 Test.2;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn ir_when_just() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x : [ Nothing, Just I64 ]
|
|
x = Just 41
|
|
|
|
when x is
|
|
Just v -> v + 0x1
|
|
Nothing -> 0x1
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.6 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.6;
|
|
|
|
procedure Test.0 ():
|
|
let Test.12 = 0i64;
|
|
let Test.11 = 41i64;
|
|
let Test.1 = Just Test.12 Test.11;
|
|
let Test.8 = 0i64;
|
|
let Test.9 = Index 0 Test.1;
|
|
let Test.10 = lowlevel Eq Test.8 Test.9;
|
|
if Test.10 then
|
|
let Test.3 = Index 1 Test.1;
|
|
let Test.5 = 1i64;
|
|
let Test.4 = CallByName Num.24 Test.3 Test.5;
|
|
ret Test.4;
|
|
else
|
|
let Test.7 = 1i64;
|
|
ret Test.7;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn one_element_tag() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x : [ Pair I64 ]
|
|
x = Pair 2
|
|
|
|
x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn guard_pattern_true() {
|
|
compiles_to_ir(
|
|
r#"
|
|
wrapper = \{} ->
|
|
when 2 is
|
|
2 if False -> 42
|
|
_ -> 0
|
|
|
|
wrapper {}
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.3):
|
|
let Test.6 = 2i64;
|
|
joinpoint Test.12:
|
|
let Test.10 = 0i64;
|
|
ret Test.10;
|
|
in
|
|
let Test.11 = 2i64;
|
|
let Test.14 = lowlevel Eq Test.11 Test.6;
|
|
if Test.14 then
|
|
joinpoint Test.8 Test.13:
|
|
if Test.13 then
|
|
let Test.7 = 42i64;
|
|
ret Test.7;
|
|
else
|
|
jump Test.12;
|
|
in
|
|
let Test.9 = false;
|
|
jump Test.8 Test.9;
|
|
else
|
|
jump Test.12;
|
|
|
|
procedure Test.0 ():
|
|
let Test.5 = Struct {};
|
|
let Test.4 = CallByName Test.1 Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn when_on_record() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when { x: 0x2 } is
|
|
{ x } -> x + 3
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.5 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.5;
|
|
|
|
procedure Test.0 ():
|
|
let Test.6 = 2i64;
|
|
let Test.1 = Index 0 Test.6;
|
|
let Test.4 = 3i64;
|
|
let Test.3 = CallByName Num.24 Test.1 Test.4;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn when_nested_maybe() {
|
|
compiles_to_ir(
|
|
r#"
|
|
Maybe a : [ Nothing, Just a ]
|
|
|
|
x : Maybe (Maybe I64)
|
|
x = Just (Just 41)
|
|
|
|
when x is
|
|
Just (Just v) -> v + 0x1
|
|
_ -> 0x1
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.0 ():
|
|
let Test.20 = 0i64;
|
|
let Test.22 = 0i64;
|
|
let Test.21 = 41i64;
|
|
let Test.19 = Just Test.22 Test.21;
|
|
let Test.2 = Just Test.20 Test.19;
|
|
joinpoint Test.16:
|
|
let Test.10 = 1i64;
|
|
ret Test.10;
|
|
in
|
|
let Test.14 = 0i64;
|
|
let Test.15 = Index 0 Test.2;
|
|
let Test.18 = lowlevel Eq Test.14 Test.15;
|
|
if Test.18 then
|
|
let Test.11 = Index 1 Test.2;
|
|
let Test.12 = 0i64;
|
|
let Test.13 = Index 0 Test.11;
|
|
let Test.17 = lowlevel Eq Test.12 Test.13;
|
|
if Test.17 then
|
|
let Test.9 = Index 1 Test.2;
|
|
let Test.5 = Index 1 Test.9;
|
|
let Test.7 = 1i64;
|
|
let Test.6 = CallByName Num.24 Test.5 Test.7;
|
|
ret Test.6;
|
|
else
|
|
jump Test.16;
|
|
else
|
|
jump Test.16;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn when_on_two_values() {
|
|
compiles_to_ir(
|
|
r#"
|
|
when Pair 2 3 is
|
|
Pair 4 3 -> 9
|
|
Pair a b -> a + b
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.7 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.7;
|
|
|
|
procedure Test.0 ():
|
|
let Test.16 = 3i64;
|
|
let Test.15 = 2i64;
|
|
let Test.4 = Struct {Test.15, Test.16};
|
|
joinpoint Test.12:
|
|
let Test.2 = Index 0 Test.4;
|
|
let Test.3 = Index 1 Test.4;
|
|
let Test.6 = CallByName Num.24 Test.2 Test.3;
|
|
ret Test.6;
|
|
in
|
|
let Test.10 = Index 1 Test.4;
|
|
let Test.11 = 3i64;
|
|
let Test.14 = lowlevel Eq Test.11 Test.10;
|
|
if Test.14 then
|
|
let Test.8 = Index 0 Test.4;
|
|
let Test.9 = 4i64;
|
|
let Test.13 = lowlevel Eq Test.9 Test.8;
|
|
if Test.13 then
|
|
let Test.5 = 9i64;
|
|
ret Test.5;
|
|
else
|
|
jump Test.12;
|
|
else
|
|
jump Test.12;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn dict() {
|
|
compiles_to_ir(
|
|
r#"
|
|
Dict.len Dict.empty
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Dict.2 ():
|
|
let Test.4 = lowlevel DictEmpty ;
|
|
ret Test.4;
|
|
|
|
procedure Dict.8 (#Attr.2):
|
|
let Test.3 = lowlevel DictSize #Attr.2;
|
|
ret Test.3;
|
|
|
|
procedure Test.0 ():
|
|
let Test.2 = CallByName Dict.2;
|
|
let Test.1 = CallByName Dict.8 Test.2;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_append_closure() {
|
|
compiles_to_ir(
|
|
r#"
|
|
myFunction = \l -> List.append l 42
|
|
|
|
myFunction [ 1, 2 ]
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure List.5 (#Attr.2, #Attr.3):
|
|
let Test.7 = lowlevel ListAppend #Attr.2 #Attr.3;
|
|
ret Test.7;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.6 = 42i64;
|
|
let Test.5 = CallByName List.5 Test.2 Test.6;
|
|
ret Test.5;
|
|
|
|
procedure Test.0 ():
|
|
let Test.8 = 1i64;
|
|
let Test.9 = 2i64;
|
|
let Test.4 = Array [Test.8, Test.9];
|
|
let Test.3 = CallByName Test.1 Test.4;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_append() {
|
|
// TODO this leaks at the moment
|
|
// ListAppend needs to decrement its arguments
|
|
compiles_to_ir(
|
|
r#"
|
|
List.append [1] 2
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure List.5 (#Attr.2, #Attr.3):
|
|
let Test.4 = lowlevel ListAppend #Attr.2 #Attr.3;
|
|
ret Test.4;
|
|
|
|
procedure Test.0 ():
|
|
let Test.5 = 1i64;
|
|
let Test.2 = Array [Test.5];
|
|
let Test.3 = 2i64;
|
|
let Test.1 = CallByName List.5 Test.2 Test.3;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_len() {
|
|
compiles_to_ir(
|
|
r#"
|
|
x = [1,2,3]
|
|
y = [ 1.0 ]
|
|
|
|
List.len x + List.len y
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure List.7 (#Attr.2):
|
|
let Test.7 = lowlevel ListLen #Attr.2;
|
|
ret Test.7;
|
|
|
|
procedure List.7 (#Attr.2):
|
|
let Test.8 = lowlevel ListLen #Attr.2;
|
|
ret Test.8;
|
|
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.6 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.6;
|
|
|
|
procedure Test.0 ():
|
|
let Test.10 = 1i64;
|
|
let Test.11 = 2i64;
|
|
let Test.12 = 3i64;
|
|
let Test.1 = Array [Test.10, Test.11, Test.12];
|
|
let Test.9 = 1f64;
|
|
let Test.2 = Array [Test.9];
|
|
let Test.4 = CallByName List.7 Test.1;
|
|
dec Test.1;
|
|
let Test.5 = CallByName List.7 Test.2;
|
|
dec Test.2;
|
|
let Test.3 = CallByName Num.24 Test.4 Test.5;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn when_joinpoint() {
|
|
compiles_to_ir(
|
|
r#"
|
|
wrapper = \{} ->
|
|
x : [ Red, White, Blue ]
|
|
x = Blue
|
|
|
|
y =
|
|
when x is
|
|
Red -> 1
|
|
White -> 2
|
|
Blue -> 3
|
|
|
|
y
|
|
|
|
wrapper {}
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.5):
|
|
let Test.2 = 0u8;
|
|
joinpoint Test.9 Test.3:
|
|
ret Test.3;
|
|
in
|
|
switch Test.2:
|
|
case 1:
|
|
let Test.10 = 1i64;
|
|
jump Test.9 Test.10;
|
|
|
|
case 2:
|
|
let Test.11 = 2i64;
|
|
jump Test.9 Test.11;
|
|
|
|
default:
|
|
let Test.12 = 3i64;
|
|
jump Test.9 Test.12;
|
|
|
|
|
|
procedure Test.0 ():
|
|
let Test.7 = Struct {};
|
|
let Test.6 = CallByName Test.1 Test.7;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn simple_if() {
|
|
compiles_to_ir(
|
|
r#"
|
|
if True then
|
|
1
|
|
else
|
|
2
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.3 = true;
|
|
if Test.3 then
|
|
let Test.4 = 1i64;
|
|
ret Test.4;
|
|
else
|
|
let Test.2 = 2i64;
|
|
ret Test.2;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn if_multi_branch() {
|
|
compiles_to_ir(
|
|
r#"
|
|
if True then
|
|
1
|
|
else if False then
|
|
2
|
|
else
|
|
3
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.6 = true;
|
|
if Test.6 then
|
|
let Test.7 = 1i64;
|
|
ret Test.7;
|
|
else
|
|
let Test.4 = false;
|
|
if Test.4 then
|
|
let Test.5 = 2i64;
|
|
ret Test.5;
|
|
else
|
|
let Test.3 = 3i64;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn when_on_result() {
|
|
compiles_to_ir(
|
|
r#"
|
|
wrapper = \{} ->
|
|
x : Result I64 I64
|
|
x = Ok 2
|
|
|
|
y =
|
|
when x is
|
|
Ok 3 -> 1
|
|
Ok _ -> 2
|
|
Err _ -> 3
|
|
y
|
|
|
|
wrapper {}
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.5):
|
|
let Test.20 = 1i64;
|
|
let Test.19 = 2i64;
|
|
let Test.2 = Ok Test.20 Test.19;
|
|
joinpoint Test.9 Test.3:
|
|
ret Test.3;
|
|
in
|
|
let Test.16 = 1i64;
|
|
let Test.17 = Index 0 Test.2;
|
|
let Test.18 = lowlevel Eq Test.16 Test.17;
|
|
if Test.18 then
|
|
let Test.13 = Index 1 Test.2;
|
|
let Test.14 = 3i64;
|
|
let Test.15 = lowlevel Eq Test.14 Test.13;
|
|
if Test.15 then
|
|
let Test.10 = 1i64;
|
|
jump Test.9 Test.10;
|
|
else
|
|
let Test.11 = 2i64;
|
|
jump Test.9 Test.11;
|
|
else
|
|
let Test.12 = 3i64;
|
|
jump Test.9 Test.12;
|
|
|
|
procedure Test.0 ():
|
|
let Test.7 = Struct {};
|
|
let Test.6 = CallByName Test.1 Test.7;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn let_with_record_pattern() {
|
|
compiles_to_ir(
|
|
r#"
|
|
{ x } = { x: 0x2, y: 3.14 }
|
|
|
|
x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.4 = 2i64;
|
|
let Test.5 = 3.14f64;
|
|
let Test.3 = Struct {Test.4, Test.5};
|
|
let Test.1 = Index 0 Test.3;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn let_with_record_pattern_list() {
|
|
compiles_to_ir(
|
|
r#"
|
|
{ x } = { x: [ 1, 3, 4 ], y: 3.14 }
|
|
|
|
x
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.6 = 1i64;
|
|
let Test.7 = 3i64;
|
|
let Test.8 = 4i64;
|
|
let Test.4 = Array [Test.6, Test.7, Test.8];
|
|
let Test.5 = 3.14f64;
|
|
let Test.3 = Struct {Test.4, Test.5};
|
|
let Test.1 = Index 0 Test.3;
|
|
decref Test.3;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn if_guard_bind_variable_false() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
wrapper = \{} ->
|
|
when 10 is
|
|
x if x == 5 -> 0
|
|
_ -> 42
|
|
|
|
wrapper {}
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Bool.5 (#Attr.2, #Attr.3):
|
|
let Test.11 = lowlevel Eq #Attr.2 #Attr.3;
|
|
ret Test.11;
|
|
|
|
procedure Test.1 (Test.3):
|
|
let Test.6 = 10i64;
|
|
joinpoint Test.8 Test.13:
|
|
if Test.13 then
|
|
let Test.7 = 0i64;
|
|
ret Test.7;
|
|
else
|
|
let Test.12 = 42i64;
|
|
ret Test.12;
|
|
in
|
|
let Test.10 = 5i64;
|
|
let Test.9 = CallByName Bool.5 Test.6 Test.10;
|
|
jump Test.8 Test.9;
|
|
|
|
procedure Test.0 ():
|
|
let Test.5 = Struct {};
|
|
let Test.4 = CallByName Test.1 Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn alias_variable() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
x = 5
|
|
y = x
|
|
|
|
3
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
let Test.3 = 3i64;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
);
|
|
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
x = 5
|
|
y = x
|
|
|
|
y
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
ret Test.1;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn branch_store_variable() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
when 0 is
|
|
1 -> 12
|
|
a -> a
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.2 = 0i64;
|
|
let Test.5 = 1i64;
|
|
let Test.6 = lowlevel Eq Test.5 Test.2;
|
|
if Test.6 then
|
|
let Test.3 = 12i64;
|
|
ret Test.3;
|
|
else
|
|
ret Test.2;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_pass_to_function() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
x : List I64
|
|
x = [1,2,3]
|
|
|
|
id : List I64 -> List I64
|
|
id = \y -> List.set y 0 0
|
|
|
|
id x
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.4 (#Attr.2, #Attr.3, #Attr.4):
|
|
let Test.11 = lowlevel ListLen #Attr.2;
|
|
let Test.9 = lowlevel NumLt #Attr.3 Test.11;
|
|
if Test.9 then
|
|
let Test.10 = lowlevel ListSet #Attr.2 #Attr.3 #Attr.4;
|
|
ret Test.10;
|
|
else
|
|
ret #Attr.2;
|
|
|
|
procedure Test.2 (Test.3):
|
|
let Test.6 = 0i64;
|
|
let Test.7 = 0i64;
|
|
let Test.5 = CallByName List.4 Test.3 Test.6 Test.7;
|
|
ret Test.5;
|
|
|
|
procedure Test.0 ():
|
|
let Test.12 = 1i64;
|
|
let Test.13 = 2i64;
|
|
let Test.14 = 3i64;
|
|
let Test.1 = Array [Test.12, Test.13, Test.14];
|
|
let Test.4 = CallByName Test.2 Test.1;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn record_optional_field_let_no_use_default() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
f = \r ->
|
|
{ x ? 10, y } = r
|
|
x + y
|
|
|
|
|
|
f { x: 4, y: 9 }
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.3 = Index 0 Test.2;
|
|
let Test.4 = Index 1 Test.2;
|
|
let Test.7 = CallByName Num.24 Test.3 Test.4;
|
|
ret Test.7;
|
|
|
|
procedure Test.0 ():
|
|
let Test.9 = 4i64;
|
|
let Test.10 = 9i64;
|
|
let Test.6 = Struct {Test.9, Test.10};
|
|
let Test.5 = CallByName Test.1 Test.6;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn record_optional_field_let_use_default() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
f = \r ->
|
|
{ x ? 10, y } = r
|
|
x + y
|
|
|
|
|
|
f { y: 9 }
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.4 = Index 0 Test.2;
|
|
let Test.3 = 10i64;
|
|
let Test.7 = CallByName Num.24 Test.3 Test.4;
|
|
ret Test.7;
|
|
|
|
procedure Test.0 ():
|
|
let Test.9 = 9i64;
|
|
let Test.5 = CallByName Test.1 Test.9;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn record_optional_field_function_no_use_default() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
f = \{ x ? 10, y } -> x + y
|
|
|
|
|
|
f { x: 4, y: 9 }
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.1 (Test.4):
|
|
let Test.2 = Index 0 Test.4;
|
|
let Test.3 = Index 1 Test.4;
|
|
let Test.2 = 10i64;
|
|
let Test.7 = CallByName Num.24 Test.2 Test.3;
|
|
ret Test.7;
|
|
|
|
procedure Test.0 ():
|
|
let Test.9 = 4i64;
|
|
let Test.10 = 9i64;
|
|
let Test.6 = Struct {Test.9, Test.10};
|
|
let Test.5 = CallByName Test.1 Test.6;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn record_optional_field_function_use_default() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
f = \{ x ? 10, y } -> x + y
|
|
|
|
|
|
f { y: 9 }
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.1 (Test.4):
|
|
let Test.3 = Index 0 Test.4;
|
|
let Test.2 = 10i64;
|
|
let Test.2 = 10i64;
|
|
let Test.7 = CallByName Num.24 Test.2 Test.3;
|
|
ret Test.7;
|
|
|
|
procedure Test.0 ():
|
|
let Test.9 = 9i64;
|
|
let Test.5 = CallByName Test.1 Test.9;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[ignore]
|
|
#[test]
|
|
fn quicksort_help() {
|
|
crate::helpers::with_larger_debug_stack(|| {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
quicksortHelp : List (Num a), I64, I64 -> List (Num a)
|
|
quicksortHelp = \list, low, high ->
|
|
if low < high then
|
|
(Pair partitionIndex partitioned) = Pair 0 []
|
|
|
|
partitioned
|
|
|> quicksortHelp low (partitionIndex - 1)
|
|
|> quicksortHelp (partitionIndex + 1) high
|
|
else
|
|
list
|
|
|
|
quicksortHelp [] 0 0
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.3 (#Attr.2, #Attr.3):
|
|
let Test.38 = lowlevel ListLen #Attr.2;
|
|
let Test.34 = lowlevel NumLt #Attr.3 Test.38;
|
|
if Test.34 then
|
|
let Test.36 = 1i64;
|
|
let Test.37 = lowlevel ListGetUnsafe #Attr.2 #Attr.3;
|
|
let Test.35 = Ok Test.36 Test.37;
|
|
ret Test.35;
|
|
else
|
|
let Test.32 = 0i64;
|
|
let Test.33 = Struct {};
|
|
let Test.31 = Err Test.32 Test.33;
|
|
ret Test.31;
|
|
|
|
procedure List.4 (#Attr.2, #Attr.3, #Attr.4):
|
|
let Test.14 = lowlevel ListLen #Attr.2;
|
|
let Test.12 = lowlevel NumLt #Attr.3 Test.14;
|
|
if Test.12 then
|
|
let Test.13 = lowlevel ListSet #Attr.2 #Attr.3 #Attr.4;
|
|
ret Test.13;
|
|
else
|
|
ret #Attr.2;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.39 = 0i64;
|
|
let Test.28 = CallByName List.3 Test.2 Test.39;
|
|
let Test.30 = 0i64;
|
|
let Test.29 = CallByName List.3 Test.2 Test.30;
|
|
let Test.7 = Struct {Test.28, Test.29};
|
|
joinpoint Test.25:
|
|
let Test.18 = Array [];
|
|
ret Test.18;
|
|
in
|
|
let Test.19 = Index 0 Test.7;
|
|
let Test.20 = 1i64;
|
|
let Test.21 = Index 0 Test.19;
|
|
let Test.27 = lowlevel Eq Test.20 Test.21;
|
|
if Test.27 then
|
|
let Test.22 = Index 1 Test.7;
|
|
let Test.23 = 1i64;
|
|
let Test.24 = Index 0 Test.22;
|
|
let Test.26 = lowlevel Eq Test.23 Test.24;
|
|
if Test.26 then
|
|
let Test.17 = Index 0 Test.7;
|
|
let Test.3 = Index 1 Test.17;
|
|
let Test.16 = Index 1 Test.7;
|
|
let Test.4 = Index 1 Test.16;
|
|
let Test.15 = 0i64;
|
|
let Test.9 = CallByName List.4 Test.2 Test.15 Test.4;
|
|
let Test.10 = 0i64;
|
|
let Test.8 = CallByName List.4 Test.9 Test.10 Test.3;
|
|
ret Test.8;
|
|
else
|
|
dec Test.2;
|
|
jump Test.25;
|
|
else
|
|
dec Test.2;
|
|
jump Test.25;
|
|
|
|
procedure Test.0 ():
|
|
let Test.40 = 1i64;
|
|
let Test.41 = 2i64;
|
|
let Test.6 = Array [Test.40, Test.41];
|
|
let Test.5 = CallByName Test.1 Test.6;
|
|
ret Test.5;
|
|
"#
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
#[test]
|
|
fn quicksort_swap() {
|
|
crate::helpers::with_larger_debug_stack(|| {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
swap = \list ->
|
|
when Pair (List.get list 0) (List.get list 0) is
|
|
Pair (Ok atI) (Ok atJ) ->
|
|
list
|
|
|> List.set 0 atJ
|
|
|> List.set 0 atI
|
|
|
|
_ ->
|
|
[]
|
|
|
|
main =
|
|
swap [ 1, 2 ]
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.3 (#Attr.2, #Attr.3):
|
|
let Test.39 = lowlevel ListLen #Attr.2;
|
|
let Test.35 = lowlevel NumLt #Attr.3 Test.39;
|
|
if Test.35 then
|
|
let Test.38 = 1i64;
|
|
let Test.37 = lowlevel ListGetUnsafe #Attr.2 #Attr.3;
|
|
let Test.36 = Ok Test.38 Test.37;
|
|
ret Test.36;
|
|
else
|
|
let Test.34 = 0i64;
|
|
let Test.33 = Struct {};
|
|
let Test.32 = Err Test.34 Test.33;
|
|
ret Test.32;
|
|
|
|
procedure List.4 (#Attr.2, #Attr.3, #Attr.4):
|
|
let Test.15 = lowlevel ListLen #Attr.2;
|
|
let Test.13 = lowlevel NumLt #Attr.3 Test.15;
|
|
if Test.13 then
|
|
let Test.14 = lowlevel ListSet #Attr.2 #Attr.3 #Attr.4;
|
|
ret Test.14;
|
|
else
|
|
ret #Attr.2;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.40 = 0i64;
|
|
let Test.30 = CallByName List.3 Test.2 Test.40;
|
|
let Test.31 = 0i64;
|
|
let Test.29 = CallByName List.3 Test.2 Test.31;
|
|
let Test.8 = Struct {Test.29, Test.30};
|
|
joinpoint Test.26:
|
|
let Test.19 = Array [];
|
|
ret Test.19;
|
|
in
|
|
let Test.23 = Index 1 Test.8;
|
|
let Test.24 = 1i64;
|
|
let Test.25 = Index 0 Test.23;
|
|
let Test.28 = lowlevel Eq Test.24 Test.25;
|
|
if Test.28 then
|
|
let Test.20 = Index 0 Test.8;
|
|
let Test.21 = 1i64;
|
|
let Test.22 = Index 0 Test.20;
|
|
let Test.27 = lowlevel Eq Test.21 Test.22;
|
|
if Test.27 then
|
|
let Test.18 = Index 0 Test.8;
|
|
let Test.4 = Index 1 Test.18;
|
|
let Test.17 = Index 1 Test.8;
|
|
let Test.5 = Index 1 Test.17;
|
|
let Test.16 = 0i64;
|
|
let Test.10 = CallByName List.4 Test.2 Test.16 Test.5;
|
|
let Test.11 = 0i64;
|
|
let Test.9 = CallByName List.4 Test.10 Test.11 Test.4;
|
|
ret Test.9;
|
|
else
|
|
dec Test.2;
|
|
jump Test.26;
|
|
else
|
|
dec Test.2;
|
|
jump Test.26;
|
|
|
|
procedure Test.0 ():
|
|
let Test.41 = 1i64;
|
|
let Test.42 = 2i64;
|
|
let Test.7 = Array [Test.41, Test.42];
|
|
let Test.6 = CallByName Test.1 Test.7;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
#[ignore]
|
|
#[test]
|
|
fn quicksort_partition_help() {
|
|
crate::helpers::with_larger_debug_stack(|| {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
partitionHelp : I64, I64, List (Num a), I64, (Num a) -> [ Pair I64 (List (Num a)) ]
|
|
partitionHelp = \i, j, list, high, pivot ->
|
|
if j < high then
|
|
when List.get list j is
|
|
Ok value ->
|
|
if value <= pivot then
|
|
partitionHelp (i + 1) (j + 1) (swap (i + 1) j list) high pivot
|
|
else
|
|
partitionHelp i (j + 1) list high pivot
|
|
|
|
Err _ ->
|
|
Pair i list
|
|
else
|
|
Pair i list
|
|
|
|
main =
|
|
partitionHelp 0 0 [] 0 0
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
#[ignore]
|
|
#[test]
|
|
fn quicksort_full() {
|
|
crate::helpers::with_larger_debug_stack(|| {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
quicksortHelp : List (Num a), I64, I64 -> List (Num a)
|
|
quicksortHelp = \list, low, high ->
|
|
if low < high then
|
|
(Pair partitionIndex partitioned) = partition low high list
|
|
|
|
partitioned
|
|
|> quicksortHelp low (partitionIndex - 1)
|
|
|> quicksortHelp (partitionIndex + 1) high
|
|
else
|
|
list
|
|
|
|
|
|
swap : I64, I64, List a -> List a
|
|
swap = \i, j, list ->
|
|
when Pair (List.get list i) (List.get list j) is
|
|
Pair (Ok atI) (Ok atJ) ->
|
|
list
|
|
|> List.set i atJ
|
|
|> List.set j atI
|
|
|
|
_ ->
|
|
[]
|
|
|
|
partition : I64, I64, List (Num a) -> [ Pair I64 (List (Num a)) ]
|
|
partition = \low, high, initialList ->
|
|
when List.get initialList high is
|
|
Ok pivot ->
|
|
when partitionHelp (low - 1) low initialList high pivot is
|
|
Pair newI newList ->
|
|
Pair (newI + 1) (swap (newI + 1) high newList)
|
|
|
|
Err _ ->
|
|
Pair (low - 1) initialList
|
|
|
|
|
|
partitionHelp : I64, I64, List (Num a), I64, (Num a) -> [ Pair I64 (List (Num a)) ]
|
|
partitionHelp = \i, j, list, high, pivot ->
|
|
if j < high then
|
|
when List.get list j is
|
|
Ok value ->
|
|
if value <= pivot then
|
|
partitionHelp (i + 1) (j + 1) (swap (i + 1) j list) high pivot
|
|
else
|
|
partitionHelp i (j + 1) list high pivot
|
|
|
|
Err _ ->
|
|
Pair i list
|
|
else
|
|
Pair i list
|
|
|
|
|
|
|
|
quicksort = \originalList ->
|
|
n = List.len originalList
|
|
quicksortHelp originalList 0 (n - 1)
|
|
|
|
main =
|
|
quicksort [1,2,3]
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
})
|
|
}
|
|
|
|
#[test]
|
|
fn factorial() {
|
|
compiles_to_ir(
|
|
r#"
|
|
factorial = \n, accum ->
|
|
when n is
|
|
0 ->
|
|
accum
|
|
|
|
_ ->
|
|
factorial (n - 1) (n * accum)
|
|
|
|
factorial 10 1
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Num.25 (#Attr.2, #Attr.3):
|
|
let Test.14 = lowlevel NumSub #Attr.2 #Attr.3;
|
|
ret Test.14;
|
|
|
|
procedure Num.26 (#Attr.2, #Attr.3):
|
|
let Test.12 = lowlevel NumMul #Attr.2 #Attr.3;
|
|
ret Test.12;
|
|
|
|
procedure Test.1 (Test.2, Test.3):
|
|
joinpoint Test.7 Test.2 Test.3:
|
|
let Test.15 = 0i64;
|
|
let Test.16 = lowlevel Eq Test.15 Test.2;
|
|
if Test.16 then
|
|
ret Test.3;
|
|
else
|
|
let Test.13 = 1i64;
|
|
let Test.10 = CallByName Num.25 Test.2 Test.13;
|
|
let Test.11 = CallByName Num.26 Test.2 Test.3;
|
|
jump Test.7 Test.10 Test.11;
|
|
in
|
|
jump Test.7 Test.2 Test.3;
|
|
|
|
procedure Test.0 ():
|
|
let Test.5 = 10i64;
|
|
let Test.6 = 1i64;
|
|
let Test.4 = CallByName Test.1 Test.5 Test.6;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn is_nil() {
|
|
compiles_to_ir(
|
|
r#"
|
|
ConsList a : [ Cons a (ConsList a), Nil ]
|
|
|
|
isNil : ConsList a -> Bool
|
|
isNil = \list ->
|
|
when list is
|
|
Nil -> True
|
|
Cons _ _ -> False
|
|
|
|
isNil (Cons 0x2 Nil)
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.3):
|
|
let Test.13 = true;
|
|
let Test.15 = Index 0 Test.3;
|
|
let Test.14 = 1i64;
|
|
let Test.16 = lowlevel Eq Test.14 Test.15;
|
|
let Test.12 = lowlevel And Test.16 Test.13;
|
|
if Test.12 then
|
|
let Test.10 = true;
|
|
ret Test.10;
|
|
else
|
|
let Test.11 = false;
|
|
ret Test.11;
|
|
|
|
let Test.6 = 0i64;
|
|
let Test.7 = 2i64;
|
|
let Test.9 = 1i64;
|
|
let Test.8 = Nil Test.9;
|
|
let Test.5 = Cons Test.6 Test.7 Test.8;
|
|
let Test.4 = CallByName Test.1 Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn has_none() {
|
|
compiles_to_ir(
|
|
r#"
|
|
Maybe a : [ Just a, Nothing ]
|
|
ConsList a : [ Cons a (ConsList a), Nil ]
|
|
|
|
hasNone : ConsList (Maybe a) -> Bool
|
|
hasNone = \list ->
|
|
when list is
|
|
Nil -> False
|
|
Cons Nothing _ -> True
|
|
Cons (Just _) xs -> hasNone xs
|
|
|
|
hasNone (Cons (Just 3) Nil)
|
|
"#,
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.3):
|
|
let Test.13 = true;
|
|
let Test.15 = Index 0 Test.3;
|
|
let Test.14 = 1i64;
|
|
let Test.16 = lowlevel Eq Test.14 Test.15;
|
|
let Test.12 = lowlevel And Test.16 Test.13;
|
|
if Test.12 then
|
|
let Test.10 = true;
|
|
ret Test.10;
|
|
else
|
|
let Test.11 = false;
|
|
ret Test.11;
|
|
|
|
let Test.6 = 0i64;
|
|
let Test.7 = 2i64;
|
|
let Test.9 = 1i64;
|
|
let Test.8 = Nil Test.9;
|
|
let Test.5 = Cons Test.6 Test.7 Test.8;
|
|
let Test.4 = CallByName Test.1 Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn mk_pair_of() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
mkPairOf = \x -> Pair x x
|
|
|
|
main =
|
|
mkPairOf [1,2,3]
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.2):
|
|
inc Test.2;
|
|
let Test.6 = Struct {Test.2, Test.2};
|
|
ret Test.6;
|
|
|
|
procedure Test.0 ():
|
|
let Test.7 = 1i64;
|
|
let Test.8 = 2i64;
|
|
let Test.9 = 3i64;
|
|
let Test.5 = Array [Test.7, Test.8, Test.9];
|
|
let Test.4 = CallByName Test.1 Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn fst() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
fst = \x, _ -> x
|
|
|
|
main =
|
|
fst [1,2,3] [3,2,1]
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.2, Test.3):
|
|
inc Test.2;
|
|
ret Test.2;
|
|
|
|
procedure Test.0 ():
|
|
let Test.11 = 1i64;
|
|
let Test.12 = 2i64;
|
|
let Test.13 = 3i64;
|
|
let Test.5 = Array [Test.11, Test.12, Test.13];
|
|
let Test.8 = 3i64;
|
|
let Test.9 = 2i64;
|
|
let Test.10 = 1i64;
|
|
let Test.6 = Array [Test.8, Test.9, Test.10];
|
|
let Test.4 = CallByName Test.1 Test.5 Test.6;
|
|
dec Test.6;
|
|
dec Test.5;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_cannot_update_inplace() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
x : List I64
|
|
x = [1,2,3]
|
|
|
|
add : List I64 -> List I64
|
|
add = \y -> List.set y 0 0
|
|
|
|
main =
|
|
List.len (add x) + List.len x
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.4 (#Attr.2, #Attr.3, #Attr.4):
|
|
let Test.22 = lowlevel ListLen #Attr.2;
|
|
let Test.20 = lowlevel NumLt #Attr.3 Test.22;
|
|
if Test.20 then
|
|
let Test.21 = lowlevel ListSet #Attr.2 #Attr.3 #Attr.4;
|
|
ret Test.21;
|
|
else
|
|
ret #Attr.2;
|
|
|
|
procedure List.7 (#Attr.2):
|
|
let Test.9 = lowlevel ListLen #Attr.2;
|
|
ret Test.9;
|
|
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.7 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.7;
|
|
|
|
procedure Test.1 ():
|
|
let Test.11 = 1i64;
|
|
let Test.12 = 2i64;
|
|
let Test.13 = 3i64;
|
|
let Test.10 = Array [Test.11, Test.12, Test.13];
|
|
ret Test.10;
|
|
|
|
procedure Test.2 (Test.3):
|
|
let Test.17 = 0i64;
|
|
let Test.18 = 0i64;
|
|
let Test.16 = CallByName List.4 Test.3 Test.17 Test.18;
|
|
ret Test.16;
|
|
|
|
procedure Test.0 ():
|
|
let Test.15 = CallByName Test.1;
|
|
let Test.14 = CallByName Test.2 Test.15;
|
|
let Test.5 = CallByName List.7 Test.14;
|
|
dec Test.14;
|
|
let Test.8 = CallByName Test.1;
|
|
let Test.6 = CallByName List.7 Test.8;
|
|
dec Test.8;
|
|
let Test.4 = CallByName Num.24 Test.5 Test.6;
|
|
ret Test.4;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn list_get() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
wrapper = \{} ->
|
|
List.get [1,2,3] 0
|
|
|
|
wrapper {}
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.3 (#Attr.2, #Attr.3):
|
|
let Test.15 = lowlevel ListLen #Attr.2;
|
|
let Test.11 = lowlevel NumLt #Attr.3 Test.15;
|
|
if Test.11 then
|
|
let Test.14 = 1i64;
|
|
let Test.13 = lowlevel ListGetUnsafe #Attr.2 #Attr.3;
|
|
let Test.12 = Ok Test.14 Test.13;
|
|
ret Test.12;
|
|
else
|
|
let Test.10 = 0i64;
|
|
let Test.9 = Struct {};
|
|
let Test.8 = Err Test.10 Test.9;
|
|
ret Test.8;
|
|
|
|
procedure Test.1 (Test.2):
|
|
let Test.16 = 1i64;
|
|
let Test.17 = 2i64;
|
|
let Test.18 = 3i64;
|
|
let Test.6 = Array [Test.16, Test.17, Test.18];
|
|
let Test.7 = 0i64;
|
|
let Test.5 = CallByName List.3 Test.6 Test.7;
|
|
dec Test.6;
|
|
ret Test.5;
|
|
|
|
procedure Test.0 ():
|
|
let Test.4 = Struct {};
|
|
let Test.3 = CallByName Test.1 Test.4;
|
|
ret Test.3;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn peano() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
Peano : [ S Peano, Z ]
|
|
|
|
three : Peano
|
|
three = S (S (S Z))
|
|
|
|
three
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.9 = 0i64;
|
|
let Test.11 = 0i64;
|
|
let Test.13 = 0i64;
|
|
let Test.14 = 1i64;
|
|
let Test.12 = Z Test.14;
|
|
let Test.10 = S Test.13 Test.12;
|
|
let Test.8 = S Test.11 Test.10;
|
|
let Test.2 = S Test.9 Test.8;
|
|
ret Test.2;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn peano1() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
Peano : [ S Peano, Z ]
|
|
|
|
three : Peano
|
|
three = S (S (S Z))
|
|
|
|
when three is
|
|
Z -> 0
|
|
S _ -> 1
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.13 = 0i64;
|
|
let Test.15 = 0i64;
|
|
let Test.17 = 0i64;
|
|
let Test.18 = 1i64;
|
|
let Test.16 = Z Test.18;
|
|
let Test.14 = S Test.17 Test.16;
|
|
let Test.12 = S Test.15 Test.14;
|
|
let Test.2 = S Test.13 Test.12;
|
|
let Test.9 = 1i64;
|
|
let Test.10 = Index 0 Test.2;
|
|
dec Test.2;
|
|
let Test.11 = lowlevel Eq Test.9 Test.10;
|
|
if Test.11 then
|
|
let Test.7 = 0i64;
|
|
ret Test.7;
|
|
else
|
|
let Test.8 = 1i64;
|
|
ret Test.8;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn peano2() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
Peano : [ S Peano, Z ]
|
|
|
|
three : Peano
|
|
three = S (S (S Z))
|
|
|
|
when three is
|
|
S (S _) -> 1
|
|
S (_) -> 0
|
|
Z -> 0
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.19 = 0i64;
|
|
let Test.21 = 0i64;
|
|
let Test.23 = 0i64;
|
|
let Test.24 = 1i64;
|
|
let Test.22 = Z Test.24;
|
|
let Test.20 = S Test.23 Test.22;
|
|
let Test.18 = S Test.21 Test.20;
|
|
let Test.2 = S Test.19 Test.18;
|
|
let Test.15 = 0i64;
|
|
let Test.16 = Index 0 Test.2;
|
|
let Test.17 = lowlevel Eq Test.15 Test.16;
|
|
if Test.17 then
|
|
let Test.11 = Index 1 Test.2;
|
|
let Test.12 = 0i64;
|
|
let Test.13 = Index 0 Test.11;
|
|
dec Test.11;
|
|
decref Test.2;
|
|
let Test.14 = lowlevel Eq Test.12 Test.13;
|
|
if Test.14 then
|
|
let Test.7 = 1i64;
|
|
ret Test.7;
|
|
else
|
|
let Test.9 = 0i64;
|
|
ret Test.9;
|
|
else
|
|
let Test.10 = 0i64;
|
|
ret Test.10;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn optional_when() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
f = \r ->
|
|
when r is
|
|
{ x: Blue, y ? 3 } -> y
|
|
{ x: Red, y ? 5 } -> y
|
|
|
|
a = f { x: Blue, y: 7 }
|
|
b = f { x: Blue }
|
|
c = f { x: Red, y: 11 }
|
|
d = f { x: Red }
|
|
|
|
a * b * c * d
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.26 (#Attr.2, #Attr.3):
|
|
let Test.17 = lowlevel NumMul #Attr.2 #Attr.3;
|
|
ret Test.17;
|
|
|
|
procedure Test.1 (Test.6):
|
|
let Test.22 = Index 1 Test.6;
|
|
let Test.23 = false;
|
|
let Test.24 = lowlevel Eq Test.23 Test.22;
|
|
if Test.24 then
|
|
let Test.8 = Index 0 Test.6;
|
|
ret Test.8;
|
|
else
|
|
let Test.10 = Index 0 Test.6;
|
|
ret Test.10;
|
|
|
|
procedure Test.1 (Test.6):
|
|
let Test.33 = Index 0 Test.6;
|
|
let Test.34 = false;
|
|
let Test.35 = lowlevel Eq Test.34 Test.33;
|
|
if Test.35 then
|
|
let Test.8 = 3i64;
|
|
ret Test.8;
|
|
else
|
|
let Test.10 = 5i64;
|
|
ret Test.10;
|
|
|
|
procedure Test.0 ():
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn nested_pattern_match() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
Maybe a : [ Nothing, Just a ]
|
|
|
|
x : Maybe (Maybe I64)
|
|
x = Just (Just 41)
|
|
|
|
when x is
|
|
Just (Just v) -> v + 0x1
|
|
_ -> 0x1
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.8 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.8;
|
|
|
|
procedure Test.0 ():
|
|
let Test.20 = 0i64;
|
|
let Test.22 = 0i64;
|
|
let Test.21 = 41i64;
|
|
let Test.19 = Just Test.22 Test.21;
|
|
let Test.2 = Just Test.20 Test.19;
|
|
joinpoint Test.16:
|
|
let Test.10 = 1i64;
|
|
ret Test.10;
|
|
in
|
|
let Test.14 = 0i64;
|
|
let Test.15 = Index 0 Test.2;
|
|
let Test.18 = lowlevel Eq Test.14 Test.15;
|
|
if Test.18 then
|
|
let Test.11 = Index 1 Test.2;
|
|
let Test.12 = 0i64;
|
|
let Test.13 = Index 0 Test.11;
|
|
let Test.17 = lowlevel Eq Test.12 Test.13;
|
|
if Test.17 then
|
|
let Test.9 = Index 1 Test.2;
|
|
let Test.5 = Index 1 Test.9;
|
|
let Test.7 = 1i64;
|
|
let Test.6 = CallByName Num.24 Test.5 Test.7;
|
|
ret Test.6;
|
|
else
|
|
jump Test.16;
|
|
else
|
|
jump Test.16;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn linked_list_length_twice() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
LinkedList a : [ Nil, Cons a (LinkedList a) ]
|
|
|
|
nil : LinkedList I64
|
|
nil = Nil
|
|
|
|
length : LinkedList a -> I64
|
|
length = \list ->
|
|
when list is
|
|
Nil -> 0
|
|
Cons _ rest -> 1 + length rest
|
|
|
|
length nil + length nil
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.14 (#Attr.2, #Attr.3):
|
|
let Test.9 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.9;
|
|
|
|
procedure Test.3 (Test.4):
|
|
let Test.15 = true;
|
|
let Test.16 = 1i64;
|
|
let Test.17 = Index 0 Test.4;
|
|
let Test.18 = lowlevel Eq Test.16 Test.17;
|
|
let Test.14 = lowlevel And Test.18 Test.15;
|
|
if Test.14 then
|
|
dec Test.4;
|
|
let Test.10 = 0i64;
|
|
ret Test.10;
|
|
else
|
|
let Test.5 = Index 2 Test.4;
|
|
dec Test.4;
|
|
let Test.12 = 1i64;
|
|
let Test.13 = CallByName Test.3 Test.5;
|
|
let Test.11 = CallByName Num.14 Test.12 Test.13;
|
|
ret Test.11;
|
|
|
|
procedure Test.0 ():
|
|
let Test.20 = 1i64;
|
|
let Test.2 = Nil Test.20;
|
|
let Test.7 = CallByName Test.3 Test.2;
|
|
let Test.8 = CallByName Test.3 Test.2;
|
|
let Test.6 = CallByName Num.14 Test.7 Test.8;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn rigids() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
swap : Nat, Nat, List a -> List a
|
|
swap = \i, j, list ->
|
|
when Pair (List.get list i) (List.get list j) is
|
|
Pair (Ok atI) (Ok atJ) ->
|
|
foo = atJ
|
|
|
|
list
|
|
|> List.set i foo
|
|
|> List.set j atI
|
|
|
|
_ ->
|
|
[]
|
|
|
|
main =
|
|
swap 0 0 [0x1]
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.3 (#Attr.2, #Attr.3):
|
|
let Test.41 = lowlevel ListLen #Attr.2;
|
|
let Test.37 = lowlevel NumLt #Attr.3 Test.41;
|
|
if Test.37 then
|
|
let Test.40 = 1i64;
|
|
let Test.39 = lowlevel ListGetUnsafe #Attr.2 #Attr.3;
|
|
let Test.38 = Ok Test.40 Test.39;
|
|
ret Test.38;
|
|
else
|
|
let Test.36 = 0i64;
|
|
let Test.35 = Struct {};
|
|
let Test.34 = Err Test.36 Test.35;
|
|
ret Test.34;
|
|
|
|
procedure List.4 (#Attr.2, #Attr.3, #Attr.4):
|
|
let Test.19 = lowlevel ListLen #Attr.2;
|
|
let Test.17 = lowlevel NumLt #Attr.3 Test.19;
|
|
if Test.17 then
|
|
let Test.18 = lowlevel ListSet #Attr.2 #Attr.3 #Attr.4;
|
|
ret Test.18;
|
|
else
|
|
ret #Attr.2;
|
|
|
|
procedure Test.1 (Test.2, Test.3, Test.4):
|
|
let Test.33 = CallByName List.3 Test.4 Test.3;
|
|
let Test.32 = CallByName List.3 Test.4 Test.2;
|
|
let Test.13 = Struct {Test.32, Test.33};
|
|
joinpoint Test.29:
|
|
let Test.22 = Array [];
|
|
ret Test.22;
|
|
in
|
|
let Test.26 = Index 1 Test.13;
|
|
let Test.27 = 1i64;
|
|
let Test.28 = Index 0 Test.26;
|
|
let Test.31 = lowlevel Eq Test.27 Test.28;
|
|
if Test.31 then
|
|
let Test.23 = Index 0 Test.13;
|
|
let Test.24 = 1i64;
|
|
let Test.25 = Index 0 Test.23;
|
|
let Test.30 = lowlevel Eq Test.24 Test.25;
|
|
if Test.30 then
|
|
let Test.21 = Index 0 Test.13;
|
|
let Test.6 = Index 1 Test.21;
|
|
let Test.20 = Index 1 Test.13;
|
|
let Test.7 = Index 1 Test.20;
|
|
let Test.15 = CallByName List.4 Test.4 Test.2 Test.7;
|
|
let Test.14 = CallByName List.4 Test.15 Test.3 Test.6;
|
|
ret Test.14;
|
|
else
|
|
dec Test.4;
|
|
jump Test.29;
|
|
else
|
|
dec Test.4;
|
|
jump Test.29;
|
|
|
|
procedure Test.0 ():
|
|
let Test.10 = 0i64;
|
|
let Test.11 = 0i64;
|
|
let Test.42 = 1i64;
|
|
let Test.12 = Array [Test.42];
|
|
let Test.9 = CallByName Test.1 Test.10 Test.11 Test.12;
|
|
ret Test.9;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn let_x_in_x() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
x = 5
|
|
|
|
answer =
|
|
1337
|
|
|
|
unused =
|
|
nested = 17
|
|
nested
|
|
|
|
answer
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
let Test.4 = 17i64;
|
|
let Test.2 = 1337i64;
|
|
ret Test.2;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn let_x_in_x_indirect() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
x = 5
|
|
|
|
answer =
|
|
1337
|
|
|
|
unused =
|
|
nested = 17
|
|
|
|
i = 1
|
|
|
|
nested
|
|
|
|
{ answer, unused }.answer
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.0 ():
|
|
let Test.1 = 5i64;
|
|
let Test.4 = 17i64;
|
|
let Test.5 = 1i64;
|
|
let Test.2 = 1337i64;
|
|
let Test.7 = Struct {Test.2, Test.4};
|
|
let Test.6 = Index 0 Test.7;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn nested_closure() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
foo = \{} ->
|
|
x = 42
|
|
f = \{} -> x
|
|
f
|
|
|
|
main =
|
|
f = foo {}
|
|
f {}
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Test.1 (Test.5):
|
|
let Test.2 = 42i64;
|
|
let Test.3 = Struct {Test.2};
|
|
ret Test.3;
|
|
|
|
procedure Test.3 (Test.9, #Attr.12):
|
|
let Test.2 = Index 0 #Attr.12;
|
|
ret Test.2;
|
|
|
|
procedure Test.0 ():
|
|
let Test.8 = Struct {};
|
|
let Test.4 = CallByName Test.1 Test.8;
|
|
let Test.7 = Struct {};
|
|
let Test.6 = CallByName Test.3 Test.7 Test.4;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn closure_in_list() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
foo = \{} ->
|
|
x = 41
|
|
|
|
f = \{} -> x
|
|
|
|
[ f ]
|
|
|
|
main =
|
|
items = foo {}
|
|
|
|
List.len items
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure List.7 (#Attr.2):
|
|
let Test.7 = lowlevel ListLen #Attr.2;
|
|
ret Test.7;
|
|
|
|
procedure Test.1 (Test.5):
|
|
let Test.2 = 41i64;
|
|
let Test.11 = Struct {Test.2};
|
|
let Test.10 = Array [Test.11];
|
|
ret Test.10;
|
|
|
|
procedure Test.3 (Test.9, #Attr.12):
|
|
let Test.2 = Index 0 #Attr.12;
|
|
ret Test.2;
|
|
|
|
procedure Test.0 ():
|
|
let Test.8 = Struct {};
|
|
let Test.4 = CallByName Test.1 Test.8;
|
|
let Test.6 = CallByName List.7 Test.4;
|
|
dec Test.4;
|
|
ret Test.6;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn somehow_drops_definitions() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
one : I64
|
|
one = 1
|
|
|
|
two : I64
|
|
two = 2
|
|
|
|
increment : I64 -> I64
|
|
increment = \x -> x + one
|
|
|
|
double : I64 -> I64
|
|
double = \x -> x * two
|
|
|
|
apply : (a -> a), a -> a
|
|
apply = \f, x -> f x
|
|
|
|
main =
|
|
apply (if True then increment else double) 42
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
fn specialize_closures() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
|
|
apply : (a -> a), a -> a
|
|
apply = \f, x -> f x
|
|
|
|
main =
|
|
one : I64
|
|
one = 1
|
|
|
|
two : I64
|
|
two = 2
|
|
|
|
b : Bool
|
|
b = True
|
|
|
|
increment : I64 -> I64
|
|
increment = \x -> x + one
|
|
|
|
double : I64 -> I64
|
|
double = \x -> if b then x * two else x
|
|
|
|
apply (if True then increment else double) 42
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
procedure Num.24 (#Attr.2, #Attr.3):
|
|
let Test.27 = lowlevel NumAdd #Attr.2 #Attr.3;
|
|
ret Test.27;
|
|
|
|
procedure Num.26 (#Attr.2, #Attr.3):
|
|
let Test.23 = lowlevel NumMul #Attr.2 #Attr.3;
|
|
ret Test.23;
|
|
|
|
procedure Test.1 (Test.2, Test.3):
|
|
let Test.15 = Index 0 Test.2;
|
|
joinpoint Test.16 Test.14:
|
|
ret Test.14;
|
|
in
|
|
switch Test.15:
|
|
case 0:
|
|
let Test.17 = CallByName Test.7 Test.3 Test.2;
|
|
jump Test.16 Test.17;
|
|
|
|
default:
|
|
let Test.18 = CallByName Test.8 Test.3 Test.2;
|
|
jump Test.16 Test.18;
|
|
|
|
|
|
procedure Test.7 (Test.9, #Attr.12):
|
|
let Test.4 = Index 1 #Attr.12;
|
|
let Test.26 = CallByName Num.24 Test.9 Test.4;
|
|
ret Test.26;
|
|
|
|
procedure Test.8 (Test.10, #Attr.12):
|
|
let Test.6 = Index 2 #Attr.12;
|
|
let Test.5 = Index 1 #Attr.12;
|
|
if Test.6 then
|
|
let Test.22 = CallByName Num.26 Test.10 Test.5;
|
|
ret Test.22;
|
|
else
|
|
ret Test.10;
|
|
|
|
procedure Test.0 ():
|
|
let Test.6 = true;
|
|
let Test.4 = 1i64;
|
|
let Test.5 = 2i64;
|
|
joinpoint Test.20 Test.12:
|
|
let Test.13 = 42i64;
|
|
let Test.11 = CallByName Test.1 Test.12 Test.13;
|
|
ret Test.11;
|
|
in
|
|
let Test.25 = true;
|
|
if Test.25 then
|
|
let Test.28 = 0i64;
|
|
let Test.7 = ClosureTag(Test.7) Test.28 Test.4;
|
|
jump Test.20 Test.7;
|
|
else
|
|
let Test.24 = 1i64;
|
|
let Test.8 = ClosureTag(Test.8) Test.24 Test.5 Test.6;
|
|
jump Test.20 Test.8;
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn specialize_lowlevel() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
|
|
apply : (a -> a), a -> a
|
|
apply = \f, x -> f x
|
|
|
|
main =
|
|
one : I64
|
|
one = 1
|
|
|
|
two : I64
|
|
two = 2
|
|
|
|
increment : I64 -> I64
|
|
increment = \x -> x + 1
|
|
|
|
double : I64 -> I64
|
|
double = \x -> x * two
|
|
|
|
when 3 is
|
|
1 -> increment 0
|
|
2 -> double 0
|
|
_ -> List.map [] (if True then increment else double) |> List.len
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
|
|
#[test]
|
|
#[ignore]
|
|
fn static_str_closure() {
|
|
compiles_to_ir(
|
|
indoc!(
|
|
r#"
|
|
app "test" provides [ main ] to "./platform"
|
|
|
|
main : Str
|
|
main =
|
|
x = "long string that is malloced"
|
|
|
|
f : {} -> Str
|
|
f = (\_ -> x)
|
|
|
|
f {}
|
|
"#
|
|
),
|
|
indoc!(
|
|
r#"
|
|
"#
|
|
),
|
|
)
|
|
}
|
|
}
|