use std::collections::BTreeMap;
use abortable_parser::iter::SliceIter;
use abortable_parser::Result as ParseResult;
use crate::ast::{
typecheck::DeriveShape, Expression, ListDef, NarrowedShape, Position, PositionedItem, Shape,
Token, TokenType, Value,
};
use crate::iter::OffsetStrIter;
use crate::parse::expression;
use crate::tokenizer::tokenize;
#[test]
fn derive_shape_values() {
let value_cases = vec![
(
Value::Boolean(PositionedItem::new(false, Position::new(0, 1, 2))),
Shape::Boolean(Position::new(0, 1, 2)),
),
(
Value::Boolean(PositionedItem::new(true, Position::new(0, 1, 2))),
Shape::Boolean(Position::new(0, 1, 2)),
),
(
Value::Int(PositionedItem::new(1, Position::new(0, 1, 2))),
Shape::Int(Position::new(0, 1, 2)),
),
(
Value::Float(PositionedItem::new(2.0, Position::new(0, 1, 2))),
Shape::Float(Position::new(0, 1, 2)),
),
(
Value::Str(PositionedItem::new("foo".into(), Position::new(0, 1, 2))),
Shape::Str(Position::new(0, 1, 2)),
),
(
Value::Tuple(PositionedItem::new(
vec![(
Token::new("foo", TokenType::BAREWORD, Position::new(0, 0, 0)),
None,
Expression::Simple(Value::Int(PositionedItem::new(3, Position::new(0, 0, 0)))),
)],
Position::new(0, 0, 0),
)),
Shape::Tuple(PositionedItem::new(
vec![(
PositionedItem::new("foo".into(), Position::new(0, 0, 0)),
Shape::Int(Position::new(0, 0, 0)),
)],
Position::new(0, 0, 0),
)),
),
(
Value::List(ListDef {
elems: vec![Expression::Simple(Value::Int(PositionedItem::new(
3,
Position::new(0, 0, 0),
)))],
pos: Position::new(0, 0, 0),
}),
Shape::List(NarrowedShape::new_with_pos(
vec![Shape::Int(Position::new(0, 0, 0))],
Position::new(0, 0, 0),
)),
),
];
for (val, shape) in value_cases {
assert_eq!(val.derive_shape(&mut BTreeMap::new()), shape);
}
}
#[test]
fn derive_shape_expressions() {
let expr_cases = vec![
("3;", Shape::Int(Position::new(1, 1, 0))),
("(3);", Shape::Int(Position::new(1, 2, 1))),
("\"foo {}\" % (1);", Shape::Str(Position::new(1, 1, 0))),
("not true;", Shape::Boolean(Position::new(1, 1, 0))),
(
"0:1;",
Shape::List(NarrowedShape::new_with_pos(
vec![Shape::Int(Position::new(1, 1, 0))],
Position::new(1, 1, 0),
)),
),
("int(\"1\");", Shape::Int(Position::new(1, 1, 0))),
("float(1);", Shape::Float(Position::new(1, 1, 0))),
("str(1);", Shape::Str(Position::new(1, 1, 0))),
("bool(\"true\");", Shape::Boolean(Position::new(1, 1, 0))),
("1 + 1;", Shape::Int(Position::new(1, 1, 0))),
];
for (expr, shape) in expr_cases {
{
let iter = OffsetStrIter::new(expr);
let tokenized = match tokenize(iter, None) {
Ok(toks) => toks,
Err(err) => panic!("failed to parse {} with error {}", expr, err),
};
let toks = SliceIter::new(&tokenized);
if let ParseResult::Complete(_, ref expr) = expression(toks) {
assert_eq!(expr.derive_shape(&mut BTreeMap::new()), shape);
} else {
assert!(false, "failed to parse expression! {:?}", expr);
};
};
}
}