weld 0.4.0

Weld is a language and runtime for improving the performance of data-intensive applications.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! Utilities used in unit testing.

use crate::ast::Expr;

pub use crate::ast::CompareIgnoringSymbols;
pub use crate::syntax::parser::parse_expr;

/// Returns a typed expression.
#[cfg(test)]
pub fn typed_expression(s: &str) -> Expr {
    use crate::ast::InferTypes;
    let mut expr = parse_expr(s).unwrap();
    expr.infer_types().unwrap();
    expr
}

/// Checks whether transform(input) == expect.
#[cfg(test)]
pub fn check_transform(input: &str, expect: &str, transform: fn(&mut Expr)) {
    use crate::ast::{PrettyPrint, PrettyPrintConfig};

    let conf = PrettyPrintConfig::default().show_types(true);

    let input = &mut typed_expression(input);
    let expect = &typed_expression(expect);

    transform(input);

    println!(
        "\nInput: {}\nExpect: {}",
        input.pretty_print_config(&conf),
        expect.pretty_print_config(&conf)
    );

    assert!(input.compare_ignoring_symbols(expect).unwrap());
}

/// Print an un-indented expression for string comparison.
#[cfg(test)]
pub fn print_expr_without_indent(e: &Expr) -> String {
    use crate::ast::{PrettyPrint, PrettyPrintConfig};
    let config = PrettyPrintConfig::default()
        .show_types(false)
        .should_indent(false);
    e.pretty_print_config(&config)
}

/// Print an un-indented expression for string comparison.
#[cfg(test)]
pub fn print_typed_expr_without_indent(e: &Expr) -> String {
    use crate::ast::{PrettyPrint, PrettyPrintConfig};
    let config = PrettyPrintConfig::default()
        .show_types(true)
        .should_indent(false);
    e.pretty_print_config(&config)
}