language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 22:09:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

isZero, isPositive, isNegative

Browse source
This commit is contained in:
Chad Stearns 2020-05-18 12:49:27 -04:00
parent 40cd77ec95
commit a01bdd66c5
6 changed files with 198 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

51
compiler/can/src/builtins.rs
View file

@ -34,9 +34,60 @@ pub fn builtin_defs(var_store: &VarStore) -> Vec<Def> {
int_rem(var_store), int_rem(var_store),
int_is_odd(var_store), int_is_odd(var_store),
int_is_even(var_store), int_is_even(var_store),
int_is_zero(var_store),
int_is_positive(var_store),
int_is_negative(var_store),
] ]
} }
/// Int.isNegative : Int -> Bool
fn int_is_negative(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;
defn(
Symbol::INT_IS_NEGATIVE,
vec![Symbol::INT_IS_NEGATIVE_ARG],
var_store,
call(
Symbol::NUM_LT,
vec![Var(Symbol::INT_IS_NEGATIVE_ARG), Int(var_store.fresh(), 0)],
var_store,
),
)
}
/// Int.isPositive : Int -> Bool
fn int_is_positive(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;
defn(
Symbol::INT_IS_POSITIVE,
vec![Symbol::INT_IS_POSITIVE_ARG],
var_store,
call(
Symbol::NUM_GT,
vec![Var(Symbol::INT_IS_POSITIVE_ARG), Int(var_store.fresh(), 0)],
var_store,
),
)
}
/// Int.isZero : Int -> Bool
fn int_is_zero(var_store: &VarStore) -> Def {
use crate::expr::Expr::*;
defn(
Symbol::INT_IS_ZERO,
vec![Symbol::INT_IS_ZERO_ARG],
var_store,
call(
Symbol::INT_EQ_I64,
vec![Var(Symbol::INT_IS_ZERO_ARG), Int(var_store.fresh(), 0)],
var_store,
),
)
}
/// Int.isOdd : Int -> Bool /// Int.isOdd : Int -> Bool
fn int_is_odd(var_store: &VarStore) -> Def { fn int_is_odd(var_store: &VarStore) -> Def {
use crate::expr::Expr::*; use crate::expr::Expr::*;

146
compiler/gen/tests/gen_builtins.rs
View file

@ -29,18 +29,24 @@ mod gen_builtins {
#[test] #[test]
fn f64_sqrt() { fn f64_sqrt() {
assert_evals_to!("Float.sqrt 144", 12.0, f64); with_larger_debug_stack(|| {
assert_evals_to!("Float.sqrt 144", 12.0, f64);
})
} }
#[test] #[test]
fn f64_round() { fn f64_round() {
assert_evals_to!("Float.round 3.6", 4, i64); with_larger_debug_stack(|| {
assert_evals_to!("Float.round 3.6", 4, i64);
})
} }
#[test] #[test]
fn f64_abs() { fn f64_abs() {
assert_evals_to!("Float.abs -4.7", 4.7, f64); with_larger_debug_stack(|| {
assert_evals_to!("Float.abs 5.8", 5.8, f64); assert_evals_to!("Float.abs -4.7", 4.7, f64);
assert_evals_to!("Float.abs 5.8", 5.8, f64);
})
} }
#[test] #[test]
@ -51,7 +57,9 @@ mod gen_builtins {
#[test] #[test]
fn empty_list_literal() { fn empty_list_literal() {
assert_evals_to!("[]", &[], &'static [i64]); with_larger_debug_stack(|| {
assert_evals_to!("[]", &[], &'static [i64]);
})
} }
#[test] #[test]
@ -61,9 +69,10 @@ mod gen_builtins {
#[test] #[test]
fn gen_if_fn() { fn gen_if_fn() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
limitedNegate = \num -> limitedNegate = \num ->
if num == 1 then if num == 1 then
-1 -1
@ -74,10 +83,11 @@ mod gen_builtins {
limitedNegate 1 limitedNegate 1
"# "#
), ),
-1, -1,
i64 i64
); );
})
} }
#[test] #[test]
@ -95,15 +105,17 @@ mod gen_builtins {
#[test] #[test]
fn gen_add_f64() { fn gen_add_f64() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
1.1 + 2.4 + 3 1.1 + 2.4 + 3
"# "#
), ),
6.5, 6.5,
f64 f64
); );
})
} }
#[test] #[test]
@ -147,15 +159,17 @@ mod gen_builtins {
#[test] #[test]
fn gen_add_i64() { fn gen_add_i64() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
1 + 2 + 3 1 + 2 + 3
"# "#
), ),
6, 6,
i64 i64
); );
})
} }
#[test] #[test]
@ -257,6 +271,26 @@ mod gen_builtins {
); );
} }
#[test]
fn gen_is_zero() {
assert_evals_to!("Int.isZero 0", true, bool);
assert_evals_to!("Int.isZero 1", false, bool);
}
#[test]
fn gen_is_positive() {
assert_evals_to!("Int.isPositive 0", false, bool);
assert_evals_to!("Int.isPositive 1", true, bool);
assert_evals_to!("Int.isPositive -5", false, bool);
}
#[test]
fn gen_is_negative() {
assert_evals_to!("Int.isNegative 0", false, bool);
assert_evals_to!("Int.isNegative 3", false, bool);
assert_evals_to!("Int.isNegative -2", false, bool);
}
#[test] #[test]
fn gen_is_odd() { fn gen_is_odd() {
assert_evals_to!("Int.isOdd 4", false, bool); assert_evals_to!("Int.isOdd 4", false, bool);
@ -387,9 +421,10 @@ mod gen_builtins {
} }
#[test] #[test]
fn tail_call_elimination() { fn tail_call_elimination() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
sum = \n, accum -> sum = \n, accum ->
when n is when n is
0 -> accum 0 -> accum
@ -397,10 +432,11 @@ mod gen_builtins {
sum 1_000_000 0 sum 1_000_000 0
"# "#
), ),
500000500000, 500000500000,
i64 i64
); );
})
} }
#[test] #[test]
fn int_negate() { fn int_negate() {
@ -425,12 +461,16 @@ mod gen_builtins {
#[test] #[test]
fn empty_list_len() { fn empty_list_len() {
assert_evals_to!("List.len []", 0, usize); with_larger_debug_stack(|| {
assert_evals_to!("List.len []", 0, usize);
})
} }
#[test] #[test]
fn basic_int_list_len() { fn basic_int_list_len() {
assert_evals_to!("List.len [ 12, 9, 6, 3 ]", 4, usize); with_larger_debug_stack(|| {
assert_evals_to!("List.len [ 12, 9, 6, 3 ]", 4, usize);
})
} }
#[test] #[test]
@ -472,37 +512,43 @@ mod gen_builtins {
#[test] #[test]
fn empty_list_is_empty() { fn empty_list_is_empty() {
assert_evals_to!("List.isEmpty []", true, bool); with_larger_debug_stack(|| {
assert_evals_to!("List.isEmpty []", true, bool);
})
} }
#[test] #[test]
fn first_int_list() { fn first_int_list() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
when List.first [ 12, 9, 6, 3 ] is when List.first [ 12, 9, 6, 3 ] is
Ok val -> val Ok val -> val
Err _ -> -1 Err _ -> -1
"# "#
), ),
12, 12,
i64 i64
); );
})
} }
#[test] #[test]
fn first_empty_list() { fn first_empty_list() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
when List.first [] is when List.first [] is
Ok val -> val Ok val -> val
Err _ -> -1 Err _ -> -1
"# "#
), ),
-1, -1,
i64 i64
); );
})
} }
#[test] #[test]

48
compiler/gen/tests/gen_primitives.rs
View file

@ -13,7 +13,7 @@ mod helpers;
#[cfg(test)] #[cfg(test)]
mod gen_primitives { mod gen_primitives {
use crate::helpers::{can_expr, infer_expr, uniq_expr, CanExprOut}; use crate::helpers::{can_expr, infer_expr, uniq_expr, with_larger_debug_stack, CanExprOut};
use bumpalo::Bump; use bumpalo::Bump;
use inkwell::context::Context; use inkwell::context::Context;
use inkwell::execution_engine::JitFunction; use inkwell::execution_engine::JitFunction;
@ -406,27 +406,30 @@ mod gen_primitives {
#[test] #[test]
fn gen_chained_defs() { fn gen_chained_defs() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
x = i1 r#"
i3 = i2 x = i1
i1 = 1337 i3 = i2
i2 = i1 i1 = 1337
y = 12.4 i2 = i1
y = 12.4
i3 i3
"# "#
), ),
1337, 1337,
i64 i64
); );
})
} }
#[test] #[test]
fn gen_nested_defs() { fn gen_nested_defs() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
x = 5 x = 5
answer = answer =
@ -457,9 +460,10 @@ mod gen_primitives {
answer answer
"# "#
), ),
1337, 1337,
i64 i64
); );
})
} }
} }

18
compiler/gen/tests/gen_tags.rs
View file

@ -13,7 +13,7 @@ mod helpers;
#[cfg(test)] #[cfg(test)]
mod gen_tags { mod gen_tags {
use crate::helpers::{can_expr, infer_expr, uniq_expr, CanExprOut}; use crate::helpers::{can_expr, infer_expr, uniq_expr, with_larger_debug_stack, CanExprOut};
use bumpalo::Bump; use bumpalo::Bump;
use inkwell::context::Context; use inkwell::context::Context;
use inkwell::execution_engine::JitFunction; use inkwell::execution_engine::JitFunction;
@ -213,9 +213,10 @@ mod gen_tags {
#[test] #[test]
fn even_odd() { fn even_odd() {
assert_evals_to!( with_larger_debug_stack(|| {
indoc!( assert_evals_to!(
r#" indoc!(
r#"
even = \n -> even = \n ->
when n is when n is
0 -> True 0 -> True
@ -230,10 +231,11 @@ mod gen_tags {
odd 5 && even 42 odd 5 && even 42
"# "#
), ),
true, true,
bool bool
); );
})
} }
#[test] #[test]

6
compiler/module/src/symbol.rs
View file

@ -623,6 +623,12 @@ define_builtins! {
29 INT_IS_ODD_ARG: "isOdd#arg" 29 INT_IS_ODD_ARG: "isOdd#arg"
30 INT_IS_EVEN: "isEven" 30 INT_IS_EVEN: "isEven"
31 INT_IS_EVEN_ARG: "isEven#arg" 31 INT_IS_EVEN_ARG: "isEven#arg"
32 INT_IS_ZERO: "isZero"
33 INT_IS_ZERO_ARG: "isZero#arg"
34 INT_IS_POSITIVE: "isPositive"
35 INT_IS_POSITIVE_ARG: "isPositive#arg"
36 INT_IS_NEGATIVE: "isNegative"
37 INT_IS_NEGATIVE_ARG: "isNegative#arg"
} }
3 FLOAT: "Float" => { 3 FLOAT: "Float" => {
0 FLOAT_FLOAT: "Float" imported // the Float.Float type alias 0 FLOAT_FLOAT: "Float" imported // the Float.Float type alias

14
compiler/solve/tests/test_uniq_solve.rs
View file

@ -116,14 +116,16 @@ mod test_uniq_solve {
#[test] #[test]
fn empty_list_literal() { fn empty_list_literal() {
infer_eq( with_larger_debug_stack(|| {
indoc!( infer_eq(
r#" indoc!(
r#"
[] []
"# "#
), ),
"Attr * (List *)", "Attr * (List *)",
); );
})
} }
#[test] #[test]