univariate_coefficient_basis.test.cpp

Go to the documentation of this file.

#include "univariate_coefficient_basis.hpp"
#include "barretenberg/ecc/curves/bn254/fr.hpp"
#include "univariate.hpp"
#include <gtest/gtest.h>
 
using namespace bb;
 
template <typename FF> class UnivariateCoefficientBasisTest : public testing::Test {
  public:
    template <size_t view_length> using UnivariateView = UnivariateView<FF, view_length>;
};
 
using FieldTypes = testing::Types<fr>;
TYPED_TEST_SUITE(UnivariateCoefficientBasisTest, FieldTypes);
 
TYPED_TEST(UnivariateCoefficientBasisTest, Conversion)
{
    fr a0 = fr::random_element();
    fr a1 = fr::random_element();
 
    Univariate<fr, 2> expected({ a0, a1 });
    UnivariateCoefficientBasis<fr, 2, true> uni_m(expected);
    Univariate<fr, 2> result(uni_m);
    EXPECT_EQ(result, expected);
}
 
TYPED_TEST(UnivariateCoefficientBasisTest, Addition)
{
    Univariate<fr, 2> f1{ { 1, 2 } };
    Univariate<fr, 2> f2{ { 3, 4 } };
    UnivariateCoefficientBasis<fr, 2, true> f1_m(f1);
    UnivariateCoefficientBasis<fr, 2, true> f2_m(f2);
 
    Univariate<fr, 2> result(f1_m + f2_m);
    Univariate<fr, 2> expected = f1 + f2;
    EXPECT_EQ(result, expected);
}
 
TYPED_TEST(UnivariateCoefficientBasisTest, Multiplication)
{
 
    Univariate<fr, 2> f1({ 1, 2 });
    Univariate<fr, 2> f2({ 3, 4 });
    UnivariateCoefficientBasis<fr, 2, true> f1_m(f1);
    UnivariateCoefficientBasis<fr, 2, true> f2_m(f2);
 
    Univariate<fr, 3> result(f1_m * f2_m);
    Univariate<fr, 3> expected = (f1.template extend_to<3>()) * (f2.template extend_to<3>());
    EXPECT_EQ(result, expected);
}