polenta 0.1.2

A toy language about polynomials over finite-fields
Documentation 
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

pub mod common;
use common::run_test_for_symbols;

#[test]
fn test_poly_powers() {
    run_test_for_symbols(
        "let P(x) = (1+(1-1))* x ^3 + (6/2 )*x^ 2 +      x + 4;",
        vec![("P", "x^3 + 3*x^2 + x + 4")],
    );
}

#[test]
fn test_const_eval() {
    run_test_for_symbols("let a = 2@7;", vec![("a", "2")]);
}

#[test]
fn test_power_precedence() {
    run_test_for_symbols("let P(x) = 3*x^2^3;", vec![("P", "3*x^8")]);
    run_test_for_symbols("let P(x) = (3*x^2)^3;", vec![("P", "27*x^6")]);
}

#[test]
fn test_pow_zero() {
    run_test_for_symbols("let a = 5^0;", vec![("a", "1")]);
    run_test_for_symbols("let P(x) = (x + 1)^0;", vec![("P", "1")]);
}

#[test]
fn test_pow_one() {
    run_test_for_symbols("let a = 7^1;", vec![("a", "7")]);
    run_test_for_symbols("let P(x) = (2*x + 3)^1;", vec![("P", "2*x + 3")]);
}

#[test]
fn test_pow_large() {
    // (x+1)^4 = x^4 + 4x^3 + 6x^2 + 4x + 1
    run_test_for_symbols(
        "let P(x) = (x + 1)^4;",
        vec![("P", "x^4 + 4*x^3 + 6*x^2 + 4*x + 1")],
    );
    // 2^10 = 1024
    run_test_for_symbols("let a = 2^10;", vec![("a", "1024")]);
}

#[test]
fn test_pow_of_zero() {
    run_test_for_symbols("let a = 0^5;", vec![("a", "0")]);
}

#[test]
fn test_let_shadowing() {
    run_test_for_symbols(
        r#"
        let P(x) = 3*x;
        let P(x) = 3*P + 5;
    "#,
        vec![("P", "9*x + 5")],
    );
}

#[test]
fn test_multiplications() {
    run_test_for_symbols(
        "let P(x) = (x + 1)*(x + 2)*(x + 4);",
        vec![("P", "x^3 + 7*x^2 + 14*x + 8")],
    );
}

#[test]
fn test_poly_eval() {
    run_test_for_symbols(
        r#"
        let P(x) = 4*x + 2; 
        let t = 2;
        let a = 5 + (P+P)@(t+t);
        "#,
        vec![("P", "4*x + 2"), ("t", "2"), ("a", "41")],
    );
}

#[test]
fn test_poly_eval_chain() {
    run_test_for_symbols(
        r#"
        let P(x) = 3*x + 1; 
        let Q(x) = x/2;
        Q@P@Q@P@3;
        "#,
        vec![("!!", "8")],
    );
}

#[test]
fn test_poly_with_identifier() {
    run_test_for_symbols(
        r#"
        let t = 2;
        let x = 5;
        let P(x) = x^t + 2*x; // x^2 + 2*x
        "#,
        vec![("t", "2"), ("x", "5"), ("P", "x^2 + 2*x")],
    );
}