roc/compiler/mono/tests/test_mono.rs

807 lines
22 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 crate::helpers::{can_expr, infer_expr, CanExprOut};
use bumpalo::Bump;
use roc_mono::layout::LayoutCache;
use roc_types::subs::Subs;
fn compiles_to_ir(src: &str, expected: &str) {
let arena = Bump::new();
let CanExprOut {
loc_expr,
var_store,
var,
constraint,
home,
mut interns,
..
} = can_expr(src);
let subs = Subs::new(var_store.into());
let mut unify_problems = Vec::new();
let (_content, mut subs) = infer_expr(subs, &mut unify_problems, &constraint, var);
// Compile and add all the Procs before adding main
let mut procs = roc_mono::ir::Procs::default();
let mut ident_ids = interns.all_ident_ids.remove(&home).unwrap();
// assume 64-bit pointers
let pointer_size = std::mem::size_of::<u64>() as u32;
// Populate Procs and Subs, and get the low-level Expr from the canonical Expr
let mut mono_problems = Vec::new();
let mut mono_env = roc_mono::ir::Env {
arena: &arena,
subs: &mut subs,
problems: &mut mono_problems,
home,
ident_ids: &mut ident_ids,
pointer_size,
jump_counter: arena.alloc(0),
};
let mut layout_cache = LayoutCache::default();
let ir_expr =
roc_mono::ir::from_can(&mut mono_env, loc_expr.value, &mut procs, &mut layout_cache);
// let mono_expr = Expr::new(&mut mono_env, loc_expr.value, &mut procs);
let procs = roc_mono::ir::specialize_all(&mut mono_env, procs, &mut LayoutCache::default());
assert_eq!(
procs.runtime_errors,
roc_collections::all::MutMap::default()
);
// Put this module's ident_ids back in the interns
interns.all_ident_ids.insert(home, ident_ids);
let mut procs_string = procs
.specialized
.iter()
.map(|(_, value)| {
if let roc_mono::ir::InProgressProc::Done(proc) = value {
proc.to_pretty(200)
} else {
String::new()
}
})
.collect::<Vec<_>>();
procs_string.push(ir_expr.to_pretty(200));
let result = procs_string.join("\n");
let the_same = result == expected;
if !the_same {
println!("{}", result);
}
assert_eq!(result, expected);
}
#[test]
fn ir_int_literal() {
compiles_to_ir(
r#"
5
"#,
indoc!(
r#"
let Test.0 = 5i64;
ret Test.0;
"#
),
)
}
#[test]
fn ir_assignment() {
compiles_to_ir(
r#"
x = 5
x
"#,
indoc!(
r#"
let Test.0 = 5i64;
ret Test.0;
"#
),
)
}
#[test]
fn ir_if() {
compiles_to_ir(
r#"
if True then 1 else 2
"#,
indoc!(
r#"
let Test.2 = true;
if Test.2 then
let Test.0 = 1i64;
jump Test.1 Test.0;
else
let Test.0 = 2i64;
jump Test.1 Test.0;
joinpoint Test.1 Test.0:
ret Test.0;
"#
),
)
}
#[test]
fn ir_when_enum() {
compiles_to_ir(
r#"
when Blue is
Red -> 1
White -> 2
Blue -> 3
"#,
indoc!(
r#"
let Test.1 = 0u8;
switch Test.1:
case 1:
let Test.3 = 1i64;
jump Test.2 Test.3;
case 2:
let Test.4 = 2i64;
jump Test.2 Test.4;
default:
let Test.5 = 3i64;
jump Test.2 Test.5;
joinpoint Test.2 Test.0:
ret Test.0;
"#
),
)
}
#[test]
fn ir_when_maybe() {
compiles_to_ir(
r#"
when Just 3 is
Just n -> n
Nothing -> 0
"#,
indoc!(
r#"
let Test.11 = 0i64;
let Test.12 = 3i64;
let Test.2 = Just Test.11 Test.12;
let Test.7 = true;
let Test.9 = Index 0 Test.2;
let Test.8 = 0i64;
let Test.10 = lowlevel Eq Test.8 Test.9;
let Test.6 = lowlevel And Test.10 Test.7;
if Test.6 then
let Test.0 = Index 1 Test.2;
jump Test.3 Test.0;
else
let Test.5 = 0i64;
jump Test.3 Test.5;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[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#"
let Test.9 = 1i64;
let Test.10 = 1i64;
let Test.11 = 2i64;
let Test.4 = These Test.9 Test.10 Test.11;
switch Test.4:
case 2:
let Test.0 = Index 1 Test.4;
jump Test.5 Test.0;
case 0:
let Test.1 = Index 1 Test.4;
jump Test.5 Test.1;
default:
let Test.2 = Index 1 Test.4;
jump Test.5 Test.2;
joinpoint Test.5 Test.3:
ret Test.3;
"#
),
)
}
#[test]
fn ir_when_record() {
compiles_to_ir(
r#"
when { x: 1, y: 3.14 } is
{ x } -> x
"#,
indoc!(
r#"
let Test.6 = 1i64;
let Test.7 = 3.14f64;
let Test.2 = Struct {Test.6, Test.7};
let Test.0 = Index 0 Test.2;
jump Test.3 Test.0;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[test]
fn ir_plus() {
compiles_to_ir(
r#"
1 + 2
"#,
indoc!(
r#"
procedure Num.14 (#Attr.2, #Attr.3):
let Test.3 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.3;
let Test.1 = 1i64;
let Test.2 = 2i64;
let Test.0 = CallByName Num.14 Test.1 Test.2;
ret Test.0;
"#
),
)
}
#[test]
fn ir_round() {
compiles_to_ir(
r#"
Num.round 3.6
"#,
indoc!(
r#"
procedure Num.36 (#Attr.2):
let Test.2 = lowlevel NumRound #Attr.2;
ret Test.2;
let Test.1 = 3.6f64;
let Test.0 = CallByName Num.36 Test.1;
ret Test.0;
"#
),
)
}
#[test]
fn ir_when_idiv() {
compiles_to_ir(
r#"
when 1000 // 10 is
Ok val -> val
Err _ -> -1
"#,
indoc!(
r#"
procedure Num.32 (#Attr.2, #Attr.3):
let Test.20 = 0i64;
let Test.17 = lowlevel NotEq #Attr.3 Test.20;
if Test.17 then
let Test.18 = 1i64;
let Test.19 = lowlevel NumDivUnchecked #Attr.2 #Attr.3;
let Test.13 = Ok Test.18 Test.19;
jump Test.14 Test.13;
else
let Test.15 = 0i64;
let Test.16 = Struct {};
let Test.13 = Err Test.15 Test.16;
jump Test.14 Test.13;
joinpoint Test.14 Test.13:
ret Test.13;
let Test.11 = 1000i64;
let Test.12 = 10i64;
let Test.2 = CallByName Num.32 Test.11 Test.12;
let Test.7 = true;
let Test.9 = Index 0 Test.2;
let Test.8 = 1i64;
let Test.10 = lowlevel Eq Test.8 Test.9;
let Test.6 = lowlevel And Test.10 Test.7;
if Test.6 then
let Test.0 = Index 1 Test.2;
jump Test.3 Test.0;
else
let Test.5 = -1i64;
jump Test.3 Test.5;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[test]
fn ir_two_defs() {
compiles_to_ir(
r#"
x = 3
y = 4
x + y
"#,
indoc!(
r#"
procedure Num.14 (#Attr.2, #Attr.3):
let Test.3 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.3;
let Test.1 = 4i64;
let Test.0 = 3i64;
let Test.2 = CallByName Num.14 Test.0 Test.1;
ret Test.2;
"#
),
)
}
#[test]
fn ir_when_just() {
compiles_to_ir(
r#"
x : [ Nothing, Just Int ]
x = Just 41
when x is
Just v -> v + 0x1
Nothing -> 0x1
"#,
indoc!(
r#"
procedure Num.14 (#Attr.2, #Attr.3):
let Test.6 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.6;
let Test.13 = 0i64;
let Test.14 = 41i64;
let Test.0 = Just Test.13 Test.14;
let Test.9 = true;
let Test.11 = Index 0 Test.0;
let Test.10 = 0i64;
let Test.12 = lowlevel Eq Test.10 Test.11;
let Test.8 = lowlevel And Test.12 Test.9;
if Test.8 then
let Test.1 = Index 1 Test.0;
let Test.5 = 1i64;
let Test.4 = CallByName Num.14 Test.1 Test.5;
jump Test.3 Test.4;
else
let Test.7 = 1i64;
jump Test.3 Test.7;
joinpoint Test.3 Test.2:
ret Test.2;
"#
),
)
}
#[test]
fn one_element_tag() {
compiles_to_ir(
r#"
x : [ Pair Int ]
x = Pair 2
x
"#,
indoc!(
r#"
let Test.2 = 2i64;
let Test.0 = Struct {Test.2};
ret Test.0;
"#
),
)
}
#[test]
fn join_points() {
compiles_to_ir(
r#"
x =
if True then 1 else 2
x
"#,
indoc!(
r#"
let Test.3 = true;
if Test.3 then
let Test.0 = 1i64;
jump Test.2 Test.0;
else
let Test.0 = 2i64;
jump Test.2 Test.0;
joinpoint Test.2 Test.0:
ret Test.0;
"#
),
)
}
#[test]
fn guard_pattern_true() {
compiles_to_ir(
r#"
when 2 is
2 if False -> 42
_ -> 0
"#,
indoc!(
r#"
let Test.1 = 2i64;
let Test.8 = true;
let Test.9 = 2i64;
let Test.12 = lowlevel Eq Test.9 Test.1;
let Test.10 = lowlevel And Test.12 Test.8;
let Test.5 = false;
jump Test.4 Test.5;
joinpoint Test.4 Test.11:
let Test.7 = lowlevel And Test.11 Test.10;
if Test.7 then
let Test.3 = 42i64;
jump Test.2 Test.3;
else
let Test.6 = 0i64;
jump Test.2 Test.6;
joinpoint Test.2 Test.0:
ret Test.0;
"#
),
)
}
#[test]
fn when_on_record() {
compiles_to_ir(
r#"
when { x: 0x2 } is
{ x } -> x + 3
"#,
indoc!(
r#"
procedure Num.14 (#Attr.2, #Attr.3):
let Test.6 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.6;
let Test.7 = 2i64;
let Test.2 = Struct {Test.7};
let Test.0 = Index 0 Test.2;
let Test.5 = 3i64;
let Test.4 = CallByName Num.14 Test.0 Test.5;
jump Test.3 Test.4;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[test]
fn when_nested_maybe() {
compiles_to_ir(
r#"
Maybe a : [ Nothing, Just a ]
x : Maybe (Maybe Int)
x = Just (Just 41)
when x is
Just (Just v) -> v + 0x1
_ -> 0x1
"#,
indoc!(
r#"
procedure Num.14 (#Attr.2, #Attr.3):
let Test.7 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.7;
let Test.19 = 0i64;
let Test.21 = 0i64;
let Test.22 = 41i64;
let Test.20 = Just Test.21 Test.22;
let Test.1 = Just Test.19 Test.20;
let Test.11 = true;
let Test.13 = Index 0 Test.1;
let Test.12 = 0i64;
let Test.18 = lowlevel Eq Test.12 Test.13;
let Test.16 = lowlevel And Test.18 Test.11;
let Test.15 = Index 0 Test.1;
let Test.14 = 0i64;
let Test.17 = lowlevel Eq Test.14 Test.15;
let Test.10 = lowlevel And Test.17 Test.16;
if Test.10 then
let Test.8 = Index 1 Test.1;
let Test.2 = Index 1 Test.8;
let Test.6 = 1i64;
let Test.5 = CallByName Num.14 Test.2 Test.6;
jump Test.4 Test.5;
else
let Test.9 = 1i64;
jump Test.4 Test.9;
joinpoint Test.4 Test.3:
ret Test.3;
"#
),
)
}
#[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.14 (#Attr.2, #Attr.3):
let Test.7 = lowlevel NumAdd #Attr.2 #Attr.3;
ret Test.7;
let Test.17 = 2i64;
let Test.18 = 3i64;
let Test.3 = Struct {Test.17, Test.18};
let Test.9 = true;
let Test.10 = 4i64;
let Test.11 = Index 0 Test.3;
let Test.16 = lowlevel Eq Test.10 Test.11;
let Test.14 = lowlevel And Test.16 Test.9;
let Test.12 = 3i64;
let Test.13 = Index 1 Test.3;
let Test.15 = lowlevel Eq Test.12 Test.13;
let Test.8 = lowlevel And Test.15 Test.14;
if Test.8 then
let Test.5 = 9i64;
jump Test.4 Test.5;
else
let Test.0 = Index 0 Test.3;
let Test.1 = Index 1 Test.3;
let Test.6 = CallByName Num.14 Test.0 Test.1;
jump Test.4 Test.6;
joinpoint Test.4 Test.2:
ret Test.2;
"#
),
)
}
#[test]
fn list_push() {
compiles_to_ir(
r#"
List.push [1] 2
"#,
indoc!(
r#"
procedure List.5 (#Attr.2, #Attr.3):
let Test.3 = lowlevel ListPush #Attr.2 #Attr.3;
ret Test.3;
let Test.4 = 1i64;
let Test.1 = Array [Test.4];
let Test.2 = 2i64;
let Test.0 = CallByName List.5 Test.1 Test.2;
ret Test.0;
"#
),
)
}
#[test]
fn when_joinpoint() {
compiles_to_ir(
r#"
x : [ Red, White, Blue ]
x = Blue
y =
when x is
Red -> 1
White -> 2
Blue -> 3
y
"#,
indoc!(
r#"
let Test.0 = 0u8;
switch Test.0:
case 1:
let Test.4 = 1i64;
jump Test.3 Test.4;
case 2:
let Test.5 = 2i64;
jump Test.3 Test.5;
default:
let Test.6 = 3i64;
jump Test.3 Test.6;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[test]
fn if_multi_branch() {
compiles_to_ir(
r#"
if True then
1
else if False then
2
else
3
"#,
indoc!(
r#"
let Test.4 = true;
if Test.4 then
let Test.0 = 1i64;
jump Test.1 Test.0;
else
let Test.3 = false;
if Test.3 then
let Test.0 = 2i64;
jump Test.2 Test.0;
else
let Test.0 = 3i64;
jump Test.2 Test.0;
joinpoint Test.2 Test.0:
jump Test.1 Test.0;
joinpoint Test.1 Test.0:
ret Test.0;
"#
),
)
}
#[test]
fn when_on_result() {
compiles_to_ir(
r#"
x : Result Int Int
x = Ok 2
y =
when x is
Ok 3 -> 1
Ok _ -> 2
Err _ -> 3
y
"#,
indoc!(
r#"
let Test.17 = 1i64;
let Test.18 = 2i64;
let Test.0 = Ok Test.17 Test.18;
let Test.13 = true;
let Test.15 = Index 0 Test.0;
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.8 = true;
let Test.9 = 3i64;
let Test.10 = Index 0 Test.0;
let Test.11 = lowlevel Eq Test.9 Test.10;
let Test.7 = lowlevel And Test.11 Test.8;
if Test.7 then
let Test.4 = 1i64;
jump Test.3 Test.4;
else
let Test.5 = 2i64;
jump Test.3 Test.5;
else
let Test.6 = 3i64;
jump Test.3 Test.6;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
#[test]
fn let_with_record_pattern() {
compiles_to_ir(
r#"
{ x } = { x: 0x2, y: 3.14 }
x
"#,
indoc!(
r#"
let Test.4 = 2i64;
let Test.5 = 3.14f64;
let Test.3 = Struct {Test.4, Test.5};
let Test.0 = Index 0 Test.3;
ret Test.0;
"#
),
)
}
#[test]
fn if_guard_bind_variable_false() {
compiles_to_ir(
indoc!(
r#"
when 10 is
x if x == 5 -> 0
_ -> 42
"#
),
indoc!(
r#"
procedure Bool.5 (#Attr.2, #Attr.3):
let Test.8 = lowlevel Eq #Attr.2 #Attr.3;
ret Test.8;
let Test.2 = 10i64;
let Test.11 = true;
let Test.0 = alias Test.2;
let Test.7 = 5i64;
let Test.6 = CallByName Bool.5 Test.0 Test.7;
jump Test.5 Test.6;
joinpoint Test.5 Test.12:
let Test.10 = lowlevel And Test.12 Test.11;
if Test.10 then
let Test.4 = 0i64;
jump Test.3 Test.4;
else
let Test.9 = 42i64;
jump Test.3 Test.9;
joinpoint Test.3 Test.1:
ret Test.1;
"#
),
)
}
}