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
|
//===- PolynomialMathTest.cpp - Polynomial math Tests ---------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "mlir/Dialect/Polynomial/IR/Polynomial.h"
#include "gtest/gtest.h"
using namespace mlir;
using namespace mlir::polynomial;
TEST(AddTest, checkSameDegreeAdditionOfIntPolynomial) {
IntPolynomial x = IntPolynomial::fromCoefficients({1, 2, 3});
IntPolynomial y = IntPolynomial::fromCoefficients({2, 3, 4});
IntPolynomial expected = IntPolynomial::fromCoefficients({3, 5, 7});
EXPECT_EQ(expected, x.add(y));
}
TEST(AddTest, checkDifferentDegreeAdditionOfIntPolynomial) {
IntMonomial term2t = IntMonomial(2, 1);
IntPolynomial x = IntPolynomial::fromMonomials({term2t}).value();
IntPolynomial y = IntPolynomial::fromCoefficients({2, 3, 4});
IntPolynomial expected = IntPolynomial::fromCoefficients({2, 5, 4});
EXPECT_EQ(expected, x.add(y));
EXPECT_EQ(expected, y.add(x));
}
TEST(AddTest, checkSameDegreeAdditionOfFloatPolynomial) {
FloatPolynomial x = FloatPolynomial::fromCoefficients({1.5, 2.5, 3.5});
FloatPolynomial y = FloatPolynomial::fromCoefficients({2.5, 3.5, 4.5});
FloatPolynomial expected = FloatPolynomial::fromCoefficients({4, 6, 8});
EXPECT_EQ(expected, x.add(y));
}
TEST(AddTest, checkDifferentDegreeAdditionOfFloatPolynomial) {
FloatPolynomial x = FloatPolynomial::fromCoefficients({1.5, 2.5});
FloatPolynomial y = FloatPolynomial::fromCoefficients({2.5, 3.5, 4.5});
FloatPolynomial expected = FloatPolynomial::fromCoefficients({4, 6, 4.5});
EXPECT_EQ(expected, x.add(y));
EXPECT_EQ(expected, y.add(x));
}
|
