waalge/aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add test for curry optimization that is a reduced case of #945

Browse Source
This commit is contained in:
microproofs
2024-05-22 15:46:32 -04:00
parent c3ad52b50f
commit 3d06129d9b
35 changed files with 111 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
crates/uplc/src/optimize/shrinker.rs
View File

@@ -244,7 +244,7 @@ impl DefaultFunction {
if let (Constant::Integer(i), Constant::ByteString(_)) =
(c.as_ref(), c2.as_ref())
{
i >= &0.into() && i < &255.into()
i >= &0.into() && i < &256.into()
} else {
false
}
@@ -2817,4 +2817,79 @@ mod tests {
compare_optimization(expected, program, |p| p.builtin_curry_reducer());
}
#[test]
fn curry_reducer_test_4() {
let program: Program<Name> = Program {
version: (1, 0, 0),
term: Term::bool(true)
.delayed_if_then_else(
Term::integer(2.into()),
Term::add_integer()
.apply(
Term::add_integer()
.apply(Term::var("x"))
.apply(Term::var("y")),
)
.apply(Term::var("z"))
.lambda("z")
.apply(
Term::index_bytearray()
.apply(Term::byte_string(vec![
1, 2, 4, 8, 16, 32, 64, 128, 255, 255,
]))
.apply(Term::integer(35.into())),
)
.lambda("y")
.apply(
Term::index_bytearray()
.apply(Term::byte_string(vec![
1, 2, 4, 8, 16, 32, 64, 128, 255, 255,
]))
.apply(Term::integer(35.into())),
),
)
.lambda("x")
.apply(
Term::bool(true).delayed_if_then_else(
Term::integer(1.into()),
Term::index_bytearray()
.apply(Term::byte_string(vec![
1, 2, 4, 8, 16, 32, 64, 128, 255, 255,
]))
.apply(Term::integer(35.into())),
),
),
};
let expected = Program {
version: (1, 0, 0),
term: Term::bool(true)
.delayed_if_then_else(
Term::integer(2.into()),
Term::add_integer()
.apply(
Term::add_integer()
.apply(Term::var("x"))
.apply(Term::var("y")),
)
.apply(Term::var("z"))
.lambda("z")
.apply(Term::var("good_curry").apply(Term::integer(35.into())))
.lambda("y")
.apply(Term::var("good_curry").apply(Term::integer(35.into()))),
)
.lambda("x")
.apply(Term::bool(true).delayed_if_then_else(
Term::integer(1.into()),
Term::var("good_curry").apply(Term::integer(35.into())),
))
.lambda("good_curry")
.apply(Term::index_bytearray().apply(Term::byte_string(vec![
1, 2, 4, 8, 16, 32, 64, 128, 255, 255,
]))),
};
compare_optimization(expected, program, |p| p.builtin_curry_reducer());
}
}