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

Add all but data parsing

Browse Source 
Updates the parsing to the standard, *except* for Data, since that'll be more involved
This commit is contained in:
Pi Lanningham
2023-07-04 01:26:03 -04:00
committed by Lucas
parent a48c45b737
commit 6d9a95ef2d
2 changed files with 130 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

238
crates/uplc/tests/pretty_tests.rs
View File

@@ -1,58 +1,64 @@
use num_bigint::ToBigInt;
use uplc::{
ast::{Constant, DeBruijn, Term, Type},
Constr, PlutusData,
ast::{Constant, Term, Type, Name},
Constr, PlutusData, parser::term,
};
// Examples sourced from https://github.com/input-output-hk/plutus/issues/4751#issuecomment-1538377273
fn round_trip(old_term: Term::<Name>, pp: &str) {
//assert_eq!(old_term.to_pretty(), pp);
let new_term = term(pp).expect("failed to parse");
assert_eq!(new_term, old_term);
}
#[test]
fn constant_list_integer() {
let term = Term::<DeBruijn>::Constant(
Constant::ProtoList(
Type::Integer.into(),
vec![
Constant::Integer(0.to_bigint().unwrap()),
Constant::Integer(1.to_bigint().unwrap()),
Constant::Integer(2.to_bigint().unwrap()),
],
)
.into(),
round_trip(
Term::<Name>::Constant(
Constant::ProtoList(
Type::Integer.into(),
vec![
Constant::Integer(0.to_bigint().unwrap()),
Constant::Integer(1.to_bigint().unwrap()),
Constant::Integer(2.to_bigint().unwrap()),
],
)
.into(),
),
"(con (list integer) [0, 1, 2])",
);
assert_eq!(term.to_pretty(), "(con (list integer) [0, 1, 2])");
}
#[test]
fn constant_pair_bool_bytestring() {
let term = Term::<DeBruijn>::Constant(
Constant::ProtoPair(
Type::Bool.into(),
Type::ByteString.into(),
Constant::Bool(true).into(),
Constant::ByteString(vec![0x01, 0x23, 0x45]).into(),
)
.into(),
);
assert_eq!(
term.to_pretty(),
"(con (pair bool bytestring) (True, #012345))"
round_trip(
Term::<Name>::Constant(
Constant::ProtoPair(
Type::Bool.into(),
Type::ByteString.into(),
Constant::Bool(true).into(),
Constant::ByteString(vec![0x01, 0x23, 0x45]).into(),
)
.into(),
),
"(con (pair bool bytestring) (True, #012345))",
);
}
#[test]
fn constant_pair_unit_string() {
let term = Term::<DeBruijn>::Constant(
Constant::ProtoPair(
Type::Unit.into(),
Type::String.into(),
Constant::Unit.into(),
Constant::String("hello universe".into()).into(),
)
.into(),
);
assert_eq!(
term.to_pretty(),
"(con (pair unit string) ((), \"hello universe\"))"
round_trip(
Term::<Name>::Constant(
Constant::ProtoPair(
Type::Unit.into(),
Type::String.into(),
Constant::Unit.into(),
Constant::String("hello universe".into()).into(),
)
.into(),
),
"(con (pair unit string) ((), \"hello universe\"))",
)
}
@@ -61,100 +67,104 @@ fn constant_deeply_nested_list() {
let t0 = Type::Integer;
let t1 = Type::List(t0.clone().into());
let t2 = Type::List(t1.clone().into());
let term = Term::<DeBruijn>::Constant(
Constant::ProtoList(
t2,
vec![
Constant::ProtoList(
t1.clone(),
vec![
Constant::ProtoList(
t0.clone(),
vec![Constant::Integer(-1.to_bigint().unwrap())],
),
Constant::ProtoList(t0.clone(), vec![]),
],
),
Constant::ProtoList(
t1.clone(),
vec![
Constant::ProtoList(t0.clone(), vec![]),
Constant::ProtoList(
t0.clone(),
vec![
Constant::Integer(2.to_bigint().unwrap()),
Constant::Integer(3.to_bigint().unwrap()),
],
),
],
),
],
)
.into(),
);
assert_eq!(
term.to_pretty(),
"(con (list (list (list integer))) [[[-1], []], [[], [2, 3]]])"
round_trip(
Term::<Name>::Constant(
Constant::ProtoList(
t2,
vec![
Constant::ProtoList(
t1.clone(),
vec![
Constant::ProtoList(
t0.clone(),
vec![Constant::Integer(-1.to_bigint().unwrap())],
),
Constant::ProtoList(t0.clone(), vec![]),
],
),
Constant::ProtoList(
t1.clone(),
vec![
Constant::ProtoList(t0.clone(), vec![]),
Constant::ProtoList(
t0.clone(),
vec![
Constant::Integer(2.to_bigint().unwrap()),
Constant::Integer(3.to_bigint().unwrap()),
],
),
],
),
],
)
.into(),
),
"(con (list (list (list integer))) [[[-1], []], [[], [2, 3]]])",
);
}
#[test]
fn constant_data_constr() {
let term = Term::<DeBruijn>::Constant(
Constant::Data(PlutusData::Constr(Constr::<PlutusData> {
tag: 1,
any_constructor: None,
fields: vec![PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(
2.into(),
))],
}))
.into(),
round_trip(
Term::<Name>::Constant(
Constant::Data(PlutusData::Constr(Constr::<PlutusData> {
tag: 1,
any_constructor: None,
fields: vec![PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(
2.into(),
))],
}))
.into(),
),
"(con data (Constr 1 [I 2]))",
);
assert_eq!(term.to_pretty(), "(con data (Constr 1 [I 2]))");
}
#[test]
fn constant_data_map() {
let term = Term::<DeBruijn>::Constant(
Constant::Data(PlutusData::Map(uplc::KeyValuePairs::Def(vec![
(
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(0.into())),
PlutusData::BoundedBytes(vec![0x00].into()),
),
(
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(1.into())),
PlutusData::BoundedBytes(vec![0x0f].into()),
),
])))
.into(),
);
assert_eq!(
term.to_pretty(),
"(con data (Map [(I 0, B #00), (I 1, B #0f)]))"
round_trip(
Term::<Name>::Constant(
Constant::Data(PlutusData::Map(uplc::KeyValuePairs::Def(vec![
(
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(0.into())),
PlutusData::BoundedBytes(vec![0x00].into()),
),
(
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(1.into())),
PlutusData::BoundedBytes(vec![0x0f].into()),
),
])))
.into(),
),
"(con data (Map [(I 0, B #00), (I 1, B #0f)]))",
);
}
#[test]
fn constant_data_list() {
let term = Term::<DeBruijn>::Constant(
Constant::Data(PlutusData::Array(vec![
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(0.into())),
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(1.into())),
]))
.into(),
round_trip(
Term::<Name>::Constant(
Constant::Data(PlutusData::Array(vec![
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(0.into())),
PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(1.into())),
]))
.into(),
),
"(con data (List [I 0, I 1]))",
);
assert_eq!(term.to_pretty(), "(con data (List [I 0, I 1]))");
}
#[test]
fn constant_data_int() {
let term = Term::<DeBruijn>::Constant(
Constant::Data(PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(
2.into(),
)))
.into(),
round_trip(
Term::<Name>::Constant(
Constant::Data(PlutusData::BigInt(pallas_primitives::alonzo::BigInt::Int(
2.into(),
)))
.into(),
),
"(con data (I 2))",
);
assert_eq!(term.to_pretty(), "(con data (I 2))");
// TODO: large integers currently encode as bytestrings, which isn't great
/*
@@ -171,8 +181,10 @@ fn constant_data_int() {
#[test]
fn constant_data_bytes() {
let term = Term::<DeBruijn>::Constant(
Constant::Data(PlutusData::BoundedBytes(vec![0x00, 0x1A].into())).into(),
round_trip(
Term::<Name>::Constant(
Constant::Data(PlutusData::BoundedBytes(vec![0x00, 0x1A].into())).into(),
),
"(con data (B #001a))",
);
assert_eq!(term.to_pretty(), "(con data (B #001a))");
}