groebner-zx/polynomial_test.cpp at main · dschepler/groebner-zx · GitHub

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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#include "polynomial.h"
#include <gtest/gtest.h>

TEST(Polynomial, Equality)
{
    EXPECT_EQ((polynomial {}), (polynomial {}));
    EXPECT_EQ((polynomial { 0, 0 }), (polynomial {}));
    EXPECT_EQ((polynomial {}), (polynomial { 0, 0 }));
    EXPECT_EQ((polynomial { 1, 2 }), (polynomial { 1, 2 }));
    EXPECT_EQ((polynomial { 0, 1, 2 }), (polynomial { 1, 2 }));
    EXPECT_EQ((polynomial { 1, 2 }), (polynomial { 0, 1, 2 }));
    EXPECT_NE((polynomial { 1, 2 }), (polynomial { 1, 2, 0 }));
}

TEST(Polynomial, Degree)
{
    EXPECT_EQ((polynomial {}.degree()), -1);
    EXPECT_EQ((polynomial { 5 }.degree()), 0);
    EXPECT_EQ((polynomial { 1, 2, 3 }.degree()), 2);
}

TEST(Polynomial, Coefficient)
{
    EXPECT_EQ((polynomial {}.coefficient(0)), 0);
    EXPECT_EQ((polynomial {}.coefficient(1)), 0);
    EXPECT_EQ((polynomial { 5 }.coefficient(0)), 5);
    EXPECT_EQ((polynomial { 5 }.coefficient(1)), 0);
    EXPECT_EQ((polynomial { 5 }.coefficient(2)), 0);
    EXPECT_EQ((polynomial { 1, 2, 3 }.coefficient(0)), 3);
    EXPECT_EQ((polynomial { 1, 2, 3 }.coefficient(1)), 2);
    EXPECT_EQ((polynomial { 1, 2, 3 }.coefficient(2)), 1);
    EXPECT_EQ((polynomial { 1, 2, 3 }.coefficient(3)), 0);
    EXPECT_EQ((polynomial { 1, 2, 3 }.coefficient(4)), 0);
}

TEST(Polynomial, Negate)
{
    EXPECT_EQ((-polynomial {}), (polynomial {}));
    EXPECT_EQ((-polynomial { 1, 2 }), (polynomial { -1, -2 }));
    EXPECT_EQ((-polynomial { 1, -3, -2 }), (polynomial { -1, 3, 2 }));
}

TEST(Polynomial, TimesXTo)
{
    EXPECT_EQ(times_x_to(polynomial {}, 5), (polynomial {}));
    EXPECT_EQ(times_x_to(polynomial { 1, 2, 3 }, 0), (polynomial { 1, 2, 3 }));
    EXPECT_EQ(times_x_to(polynomial { 1, 2, 3 }, 3),
        (polynomial { 1, 2, 3, 0, 0, 0 }));
}

TEST(Polynomial, ScalarMult)
{
    EXPECT_EQ((2 * polynomial {}), (polynomial {}));
    EXPECT_EQ((0 * polynomial { 1, 2, 3 }), (polynomial {}));
    EXPECT_EQ((3 * polynomial { 1, -2, 3 }), (polynomial { 3, -6, 9 }));
    EXPECT_EQ((-3 * polynomial { 1, -2, 3 }), (polynomial { -3, 6, -9 }));
}

TEST(Polynomial, Add)
{
    EXPECT_EQ((polynomial {} + polynomial {}), (polynomial {}));
    EXPECT_EQ((polynomial { 1, 2, 3 } + polynomial {}), (polynomial { 1, 2, 3 }));
    EXPECT_EQ((polynomial {} + polynomial { 1, 2, 3 }), (polynomial { 1, 2, 3 }));
    EXPECT_EQ((polynomial { 1, 2, 3 } + polynomial { 1, 2, 3 }), (polynomial { 2, 4, 6 }));
    EXPECT_EQ((polynomial { 1, 2, 3 } + polynomial { -1, -2, -3 }), (polynomial {}));
    EXPECT_EQ((polynomial { 1, 2, 3, 4 } + polynomial { -1, -2, 5, 6 }),
        (polynomial { 8, 10 }));
    EXPECT_EQ((polynomial { 1, 2, 3, 4 } + polynomial { 5, 6 }),
        (polynomial { 1, 2, 8, 10 }));
    EXPECT_EQ((polynomial { 5, 6 } + polynomial { 1, 2, 3, 4 }),
        (polynomial { 1, 2, 8, 10 }));
}

TEST(Polynomial, Subtract)
{
    EXPECT_EQ((polynomial {} - polynomial {}), (polynomial {}));
    EXPECT_EQ((polynomial { 1, 2, 3 } - polynomial {}), (polynomial { 1, 2, 3 }));
    EXPECT_EQ((polynomial {} - polynomial { 1, 2, 3 }), (polynomial { -1, -2, -3 }));
    EXPECT_EQ((polynomial { 1, 2, 3 } - polynomial { 1, 2, 3 }), (polynomial {}));
    EXPECT_EQ((polynomial { 1, 2, 3, 4 } - polynomial { 1, 2, 5, 6 }),
        (polynomial { -2, -2 }));
    EXPECT_EQ((polynomial { 1, 2, 3, 4 } - polynomial { 5, 6 }),
        (polynomial { 1, 2, -2, -2 }));
    EXPECT_EQ((polynomial { 5, 6 } - polynomial { 1, 2, 3, 4 }),
        (polynomial { -1, -2, 2, 2 }));
}

TEST(Polynomial, Multiply)
{
    EXPECT_EQ((polynomial {} * polynomial {}), (polynomial {}));
    EXPECT_EQ((polynomial {} * polynomial { 1, 2, 3 }), (polynomial {}));
    EXPECT_EQ((polynomial { 1, 2, 3 } * polynomial {}), (polynomial {}));
    EXPECT_EQ((polynomial { 2 } * polynomial { 1, 2, 3 }), (polynomial { 2, 4, 6 }));
    EXPECT_EQ((polynomial { 1, 2, 3 } * polynomial { 2 }), (polynomial { 2, 4, 6 }));
    EXPECT_EQ((polynomial { 1, 5 } * polynomial { 1, 6 }), (polynomial { 1, 11, 30 }));
    EXPECT_EQ((polynomial { 1, 2 } * polynomial { 1, -2 }), (polynomial { 1, 0, -4 }));
    EXPECT_EQ((polynomial { 1, 1, 1, 1, 1 } * polynomial { 1, 1, 1 }), (polynomial { 1, 2, 3, 3, 3, 2, 1 }));
}

TEST(Polynomial, ExprTemplates)
{
    polynomial p { 2, 3, 4 };
    polynomial q { 5, 6 };
    EXPECT_EQ(p * q - p, p * (q - polynomial { 1 }));
    EXPECT_EQ(p + q + p, 2 * p + q);
    EXPECT_EQ(3 * p + 4 * times_x_to(q, 2), (polynomial { 20, 30, 9, 12 }));
    EXPECT_EQ(5 * p - 2 * times_x_to(q, 1), (polynomial { 3, 20 }));
}

TEST(Polynomial, ToString)
{
    EXPECT_EQ((polynomial {}.to_string()), "0");
    EXPECT_EQ((polynomial { 13 }.to_string()), "13");
    EXPECT_EQ((polynomial { -3 }.to_string()), "-3");
    EXPECT_EQ((polynomial { 1, 0 }.to_string()), "x");
    EXPECT_EQ((polynomial { 1, 4 }.to_string()), "x + 4");
    EXPECT_EQ((polynomial { 1, -4 }.to_string()), "x - 4");
    EXPECT_EQ((polynomial { 2, 3 }.to_string()), "2 x + 3");
    EXPECT_EQ((polynomial { -2, -3 }.to_string()), "-2 x - 3");
    EXPECT_EQ((polynomial { -1, 5 }.to_string()), "-x + 5");
    EXPECT_EQ((polynomial { 1, 3, 2 }.to_string()), "x^2 + 3 x + 2");
    EXPECT_EQ((polynomial { 1, 0, -3, 0, 0, 1, 5 }.to_string()), "x^6 - 3 x^4 + x + 5");
    EXPECT_EQ((polynomial { -1, 0, 0, 0 }.to_string()), "-x^3");
    EXPECT_EQ((polynomial { 1, 0, 1 }.to_string()), "x^2 + 1");
    EXPECT_EQ((polynomial { 1, 0, -1 }.to_string()), "x^2 - 1");
    EXPECT_EQ((polynomial { -1 }.to_string()), "-1");
}