Initial import of libdpf.
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
commit
e4e666f459
4563 changed files with 1690372 additions and 0 deletions
235
test/tests/eval_full_multi_test.cpp
Normal file
235
test/tests/eval_full_multi_test.cpp
Normal file
|
|
@ -0,0 +1,235 @@
|
|||
#include <gtest/gtest.h>
|
||||
|
||||
#include "dpf.hpp"
|
||||
|
||||
#include <type_traits>
|
||||
|
||||
namespace
|
||||
{
|
||||
|
||||
template <typename A, typename B>
|
||||
auto recon(const A & a, const B & b)
|
||||
{
|
||||
if constexpr (dpf::is_secret_share_v<std::decay_t<A>>
|
||||
&& dpf::is_secret_share_v<std::decay_t<B>>)
|
||||
return dpf::reconstruct(a, b);
|
||||
else
|
||||
return a - b;
|
||||
}
|
||||
|
||||
} // namespace
|
||||
|
||||
#include "helpers/eval_common_multi_data.hpp"
|
||||
|
||||
template <typename T>
|
||||
struct EvalFullMultiTest : public testing::Test
|
||||
{
|
||||
public:
|
||||
using input_type = typename std::tuple_element_t<0, T>;
|
||||
using output_type0 = typename std::tuple_element_t<1, T>;
|
||||
using output_type1 = typename std::tuple_element_t<2, T>;
|
||||
using output_type2 = typename std::tuple_element_t<3, T>;
|
||||
using output_type3 = typename std::tuple_element_t<4, T>;
|
||||
using dpf_type = dpf::utils::dpf_type_t<dpf::prg::aes128, dpf::prg::aes128, input_type, output_type0, output_type1, output_type2, output_type3>;
|
||||
|
||||
protected:
|
||||
EvalFullMultiTest()
|
||||
: params{std::get<std::vector<T>>(allParams)},
|
||||
range{std::size_t(1) << dpf::utils::bitlength_of_v<input_type>},
|
||||
zero_output0{from_integral_type_output0(0)},
|
||||
zero_output1{from_integral_type_output1(0)},
|
||||
zero_output2{from_integral_type_output2(0)},
|
||||
zero_output3{from_integral_type_output3(0)}
|
||||
{ }
|
||||
|
||||
void SetUp() override
|
||||
{ }
|
||||
|
||||
void TearDown() override
|
||||
{ }
|
||||
|
||||
template <typename IterableT0, typename IterableT1>
|
||||
void assert_wrapper(const input_type & x, const output_type0 & y0,
|
||||
const output_type1 & y1, const output_type2 & y2, const output_type3 & y3,
|
||||
IterableT0 & iter0, IterableT1 & iter1)
|
||||
{
|
||||
auto zip0 = dpf::tuple_as_zip(iter0);
|
||||
auto zip1 = dpf::tuple_as_zip(iter1);
|
||||
auto it0 = std::cbegin(zip0);
|
||||
auto it1 = std::cbegin(zip1);
|
||||
input_type cur = std::numeric_limits<input_type>::min();
|
||||
for (std::size_t i = 0; i < range; ++i, cur = next_domain_point(cur), ++it0, ++it1)
|
||||
{
|
||||
if (cur == x)
|
||||
{
|
||||
ASSERT_EQ(recon(std::get<0>(*it0), std::get<0>(*it1)), y0);
|
||||
ASSERT_EQ(recon(std::get<1>(*it0), std::get<1>(*it1)), y1);
|
||||
ASSERT_EQ(recon(std::get<2>(*it0), std::get<2>(*it1)), y2);
|
||||
ASSERT_EQ(recon(std::get<3>(*it0), std::get<3>(*it1)), y3);
|
||||
}
|
||||
else
|
||||
{
|
||||
ASSERT_EQ(recon(std::get<0>(*it0), std::get<0>(*it1)), zero_output0);
|
||||
ASSERT_EQ(recon(std::get<1>(*it0), std::get<1>(*it1)), zero_output1);
|
||||
ASSERT_EQ(recon(std::get<2>(*it0), std::get<2>(*it1)), zero_output2);
|
||||
ASSERT_EQ(recon(std::get<3>(*it0), std::get<3>(*it1)), zero_output3);
|
||||
}
|
||||
}
|
||||
ASSERT_EQ(it0, std::end(zip0));
|
||||
ASSERT_EQ(it1, std::end(zip1));
|
||||
}
|
||||
|
||||
static constexpr auto from_integral_type = dpf::utils::make_from_integral_value<input_type>{};
|
||||
static constexpr auto from_integral_type_output0 = dpf::utils::make_from_integral_value<output_type0>{};
|
||||
static constexpr auto from_integral_type_output1 = dpf::utils::make_from_integral_value<output_type1>{};
|
||||
static constexpr auto from_integral_type_output2 = dpf::utils::make_from_integral_value<output_type2>{};
|
||||
static constexpr auto from_integral_type_output3 = dpf::utils::make_from_integral_value<output_type3>{};
|
||||
|
||||
std::vector<T> params;
|
||||
std::size_t range;
|
||||
output_type0 zero_output0;
|
||||
output_type1 zero_output1;
|
||||
output_type2 zero_output2;
|
||||
output_type3 zero_output3;
|
||||
};
|
||||
|
||||
TYPED_TEST_SUITE_P(EvalFullMultiTest);
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, Basic)
|
||||
{
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto [buf0, iter0] = dpf::eval_full<0, 1, 2, 3>(dpf0);
|
||||
auto [buf1, iter1] = dpf::eval_full<0, 1, 2, 3>(dpf1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, Outbuf)
|
||||
{
|
||||
using dpf_type = typename TestFixture::dpf_type;
|
||||
auto buf0 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>(),
|
||||
buf1 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>();
|
||||
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto iter0 = dpf::eval_full<0, 1, 2, 3>(dpf0, buf0),
|
||||
iter1 = dpf::eval_full<0, 1, 2, 3>(dpf1, buf1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, BasicFullMemoizer)
|
||||
{
|
||||
using dpf_type = typename TestFixture::dpf_type;
|
||||
auto memo0 = dpf::make_basic_full_memoizer<dpf_type>(),
|
||||
memo1 = dpf::make_basic_full_memoizer<dpf_type>();
|
||||
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto [buf0, iter0] = dpf::eval_full<0, 1, 2, 3>(dpf0, memo0);
|
||||
auto [buf1, iter1] = dpf::eval_full<0, 1, 2, 3>(dpf1, memo1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, FullTreeFullMemoizer)
|
||||
{
|
||||
using dpf_type = typename TestFixture::dpf_type;
|
||||
auto memo0 = dpf::make_full_tree_full_memoizer<dpf_type>(),
|
||||
memo1 = dpf::make_full_tree_full_memoizer<dpf_type>();
|
||||
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto [buf0, iter0] = dpf::eval_full<0, 1, 2, 3>(dpf0, memo0);
|
||||
auto [buf1, iter1] = dpf::eval_full<0, 1, 2, 3>(dpf1, memo1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, BasicFullMemoizerOutbuf)
|
||||
{
|
||||
using dpf_type = typename TestFixture::dpf_type;
|
||||
auto buf0 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>(),
|
||||
buf1 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>();
|
||||
auto memo0 = dpf::make_basic_full_memoizer<dpf_type>(),
|
||||
memo1 = dpf::make_basic_full_memoizer<dpf_type>();
|
||||
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto iter0 = dpf::eval_full<0, 1, 2, 3>(dpf0, buf0, memo0),
|
||||
iter1 = dpf::eval_full<0, 1, 2, 3>(dpf1, buf1, memo1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
TYPED_TEST_P(EvalFullMultiTest, FullTreeFullMemoizerOutbuf)
|
||||
{
|
||||
using dpf_type = typename TestFixture::dpf_type;
|
||||
auto buf0 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>(),
|
||||
buf1 = dpf::make_output_buffer_for_full<dpf_type, 0, 1, 2, 3>();
|
||||
auto memo0 = dpf::make_full_tree_full_memoizer<dpf_type>(),
|
||||
memo1 = dpf::make_full_tree_full_memoizer<dpf_type>();
|
||||
|
||||
for (auto [x, y0, y1, y2, y3] : this->params)
|
||||
{
|
||||
auto [dpf0, dpf1] = dpf::make_dpf(x, y0, y1, y2, y3);
|
||||
auto iter0 = dpf::eval_full<0, 1, 2, 3>(dpf0, buf0, memo0),
|
||||
iter1 = dpf::eval_full<0, 1, 2, 3>(dpf1, buf1, memo1);
|
||||
|
||||
this->assert_wrapper(x, y0, y1, y2, y3, iter0, iter1);
|
||||
}
|
||||
}
|
||||
|
||||
REGISTER_TYPED_TEST_SUITE_P(EvalFullMultiTest,
|
||||
Basic,
|
||||
Outbuf,
|
||||
BasicFullMemoizer,
|
||||
FullTreeFullMemoizer,
|
||||
BasicFullMemoizerOutbuf,
|
||||
FullTreeFullMemoizerOutbuf);
|
||||
using Types = testing::Types
|
||||
<
|
||||
// base test
|
||||
test_type<uint16_t, uint64_t>,
|
||||
|
||||
// test input types
|
||||
test_type<int16_t, uint64_t>,
|
||||
test_type<uint8_t, uint64_t>,
|
||||
test_type<dpf::bitstring<10>, uint64_t>,
|
||||
test_type<dpf::keyword<3, dpf::alphabets::hex>, uint64_t>,
|
||||
test_type<dpf::modint<10>, uint64_t>,
|
||||
test_type<dpf::xor_wrapper<int16_t>, uint64_t>,
|
||||
test_type<dpf::xor_wrapper<uint16_t>, uint64_t>,
|
||||
|
||||
// test output types
|
||||
test_type<uint16_t, int64_t>,
|
||||
test_type<uint16_t, uint8_t>,
|
||||
test_type<uint16_t, simde_int128>,
|
||||
test_type<uint16_t, simde_uint128>,
|
||||
test_type<uint16_t, dpf::bit>,
|
||||
test_type<uint16_t, dpf::bitstring<20, uint8_t>>,
|
||||
test_type<uint16_t, dpf::bitstring<150>>,
|
||||
test_type<uint16_t, dpf::xor_wrapper<int64_t>>,
|
||||
test_type<uint16_t, dpf::xor_wrapper<uint64_t>>,
|
||||
|
||||
// custom types
|
||||
test_type<custom_input_type, uint64_t>,
|
||||
test_type<uint16_t, custom_output_type_small>,
|
||||
test_type<uint16_t, custom_output_type_large_plus_minus>,
|
||||
test_type<uint16_t, custom_output_type_large_xor>,
|
||||
|
||||
// distinct output types
|
||||
multi_test_type<uint16_t, uint32_t, dpf::xor_wrapper<uint32_t>, dpf::bitstring<20, uint8_t>, dpf::bitstring<32>>
|
||||
>;
|
||||
INSTANTIATE_TYPED_TEST_SUITE_P(EvalFullMultiTestInstantiation, EvalFullMultiTest, Types);
|
||||
Loading…
Add table
Add a link
Reference in a new issue