1 //===- PolynomialMathTest.cpp - Polynomial math Tests ---------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "mlir/Dialect/Polynomial/IR/Polynomial.h"
10 #include "gtest/gtest.h"
11
12 using namespace mlir;
13 using namespace mlir::polynomial;
14
TEST(AddTest,checkSameDegreeAdditionOfIntPolynomial)15 TEST(AddTest, checkSameDegreeAdditionOfIntPolynomial) {
16 IntPolynomial x = IntPolynomial::fromCoefficients({1, 2, 3});
17 IntPolynomial y = IntPolynomial::fromCoefficients({2, 3, 4});
18 IntPolynomial expected = IntPolynomial::fromCoefficients({3, 5, 7});
19 EXPECT_EQ(expected, x.add(y));
20 }
21
TEST(AddTest,checkDifferentDegreeAdditionOfIntPolynomial)22 TEST(AddTest, checkDifferentDegreeAdditionOfIntPolynomial) {
23 IntMonomial term2t = IntMonomial(2, 1);
24 IntPolynomial x = IntPolynomial::fromMonomials({term2t}).value();
25 IntPolynomial y = IntPolynomial::fromCoefficients({2, 3, 4});
26 IntPolynomial expected = IntPolynomial::fromCoefficients({2, 5, 4});
27 EXPECT_EQ(expected, x.add(y));
28 EXPECT_EQ(expected, y.add(x));
29 }
30
TEST(AddTest,checkSameDegreeAdditionOfFloatPolynomial)31 TEST(AddTest, checkSameDegreeAdditionOfFloatPolynomial) {
32 FloatPolynomial x = FloatPolynomial::fromCoefficients({1.5, 2.5, 3.5});
33 FloatPolynomial y = FloatPolynomial::fromCoefficients({2.5, 3.5, 4.5});
34 FloatPolynomial expected = FloatPolynomial::fromCoefficients({4, 6, 8});
35 EXPECT_EQ(expected, x.add(y));
36 }
37
TEST(AddTest,checkDifferentDegreeAdditionOfFloatPolynomial)38 TEST(AddTest, checkDifferentDegreeAdditionOfFloatPolynomial) {
39 FloatPolynomial x = FloatPolynomial::fromCoefficients({1.5, 2.5});
40 FloatPolynomial y = FloatPolynomial::fromCoefficients({2.5, 3.5, 4.5});
41 FloatPolynomial expected = FloatPolynomial::fromCoefficients({4, 6, 4.5});
42 EXPECT_EQ(expected, x.add(y));
43 EXPECT_EQ(expected, y.add(x));
44 }
45