ultra_relation_consistency.test.cpp

Go to the documentation of this file.

 
#include "barretenberg/ecc/curves/bn254/fr.hpp"
#include "barretenberg/ecc/curves/grumpkin/grumpkin.hpp"
#include "barretenberg/flavor/generated/ultra_flavor_generated.hpp"
#include "barretenberg/relations/delta_range_constraint_relation.hpp"
#include "barretenberg/relations/elliptic_relation.hpp"
#include "barretenberg/relations/memory_relation.hpp"
#include "barretenberg/relations/non_native_field_relation.hpp"
#include "barretenberg/relations/permutation_relation.hpp"
#include "barretenberg/relations/poseidon2_external_relation.hpp"
#include "barretenberg/relations/poseidon2_internal_relation.hpp"
#include "barretenberg/relations/relation_parameters.hpp"
#include "barretenberg/relations/ultra_arithmetic_relation.hpp"
#include <gtest/gtest.h>
 
using namespace bb;
 
using FF = fr;
 
// Test inputs: the generator-emitted AllEntities<FF> for UltraFlavor — std::array<FF,
// NUM_ALL_ENTITIES> storage with EntityId-keyed access and named-method accessors. Replaces a
// hand-written 45-tuple mock that pre-dated the logderiv-lookup arithmetisation.
using InputElements = UltraFlavor_Generated::AllEntities<FF>;
 
static InputElements get_random_inputs()
{
    InputElements result;
    for (FF& slot : result.data) {
        slot = FF::random_element();
    }
    return result;
}
 
static InputElements get_special_inputs() // use non-random values: data[i] = i + 1
{
    InputElements result;
    FF idx = 0;
    for (FF& slot : result.data) {
        idx += FF(1);
        slot = idx;
    }
    return result;
}
 
class UltraRelationConsistency : public testing::Test {
  public:
    template <typename Relation>
    static void validate_relation_execution(
        const typename Relation::SumcheckArrayOfValuesOverSubrelations& expected_values,
        const InputElements& input_elements,
        const auto& parameters)
    {
        typename Relation::SumcheckArrayOfValuesOverSubrelations accumulator;
        std::fill(accumulator.begin(), accumulator.end(), FF(0));
        Relation::accumulate(accumulator, input_elements, parameters, 1);
        EXPECT_EQ(accumulator, expected_values);
    };
};
 
TEST_F(UltraRelationConsistency, ArithmeticRelation)
{
    const auto run_test = [](bool random_inputs, const FF& q_arith_value = FF::random_element()) {
        using Relation = ArithmeticRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
        const auto& w_1 = input_elements.w_l();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_4_shift = input_elements.w_4_shift();
        const auto& q_m = input_elements.q_m();
        const auto& q_l = input_elements.q_l();
        const auto& q_r = input_elements.q_r();
        const auto& q_o = input_elements.q_o();
        const auto& q_4 = input_elements.q_4();
        const auto& q_c = input_elements.q_c();
 
        // Set specific q_arith value to enable testing different modes of the arithmetic relation
        input_elements[InputElements::EntityId::q_arith] = q_arith_value;
        const auto& q_arith = input_elements.q_arith();
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
        static const FF neg_half = FF(-2).invert();
 
        FF contribution_1 = FF(0);
        FF contribution_2 = FF(0);
        if (q_arith == FF(1)) {
            // Contribution 1
            contribution_1 = (q_m * w_2 * w_1) + (q_l * w_1) + (q_r * w_2) + (q_o * w_3) + (q_4 * w_4) + q_c;
 
            // Contribution 2: None
        } else if (q_arith == FF(2)) {
            // Contribution 1
            contribution_1 = (q_m * w_2 * w_1);
            contribution_1 += ((q_l * w_1) + (q_r * w_2) + (q_o * w_3) + (q_4 * w_4) + w_4_shift + q_c) * FF(2);
 
            // Contribution 2: None
        } else if (q_arith == FF(3)) {
            // Contribution 1
            contribution_1 = (q_l * w_1) + (q_r * w_2) + (q_o * w_3) + (q_4 * w_4) + q_c;
            contribution_1 += w_4_shift * FF(2);
            contribution_1 *= FF(3);
 
            // Contribution 2
            contribution_2 = (w_1 + w_4 - w_1_shift + q_m) * FF(6);
        } else {
            // Contribution 1
            contribution_1 = (q_arith - 3) * (q_m * w_2 * w_1) * neg_half;
            contribution_1 += (q_l * w_1) + (q_r * w_2) + (q_o * w_3) + (q_4 * w_4) + q_c;
            contribution_1 += (q_arith - 1) * w_4_shift;
            contribution_1 *= q_arith;
 
            // Contribution 2
            contribution_2 = (w_1 + w_4 - w_1_shift + q_m);
            contribution_2 *= (q_arith - 2) * (q_arith - 1) * q_arith;
        }
 
        expected_values[0] = contribution_1;
        expected_values[1] = contribution_2;
 
        const auto parameters = RelationParameters<FF>::get_random();
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
    run_test(/*random_inputs=*/true, /*q_arith_value=*/FF(1));
    run_test(/*random_inputs=*/true, /*q_arith_value=*/FF(2));
    run_test(/*random_inputs=*/true, /*q_arith_value=*/FF(3));
};
 
TEST_F(UltraRelationConsistency, UltraPermutationRelation)
{
    const auto run_test = [](bool random_inputs) {
        using Relation = UltraPermutationRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& sigma_1 = input_elements.sigma_1();
        const auto& sigma_2 = input_elements.sigma_2();
        const auto& sigma_3 = input_elements.sigma_3();
        const auto& sigma_4 = input_elements.sigma_4();
        const auto& id_1 = input_elements.id_1();
        const auto& id_2 = input_elements.id_2();
        const auto& id_3 = input_elements.id_3();
        const auto& id_4 = input_elements.id_4();
        const auto& z_perm = input_elements.z_perm();
        const auto& z_perm_shift = input_elements.z_perm_shift();
        const auto& lagrange_first = input_elements.lagrange_first();
        const auto& lagrange_last = input_elements.lagrange_last();
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        const auto parameters = RelationParameters<FF>::get_random();
        const auto& beta = parameters.beta;
        const auto& gamma = parameters.gamma;
        const auto& public_input_delta = parameters.public_input_delta;
 
        // Contribution 1
        auto contribution_1 = (z_perm + lagrange_first) * (w_1 + id_1 * beta + gamma) * (w_2 + id_2 * beta + gamma) *
                                  (w_3 + id_3 * beta + gamma) * (w_4 + id_4 * beta + gamma) -
                              (z_perm_shift + lagrange_last * public_input_delta) * (w_1 + sigma_1 * beta + gamma) *
                                  (w_2 + sigma_2 * beta + gamma) * (w_3 + sigma_3 * beta + gamma) *
                                  (w_4 + sigma_4 * beta + gamma);
        expected_values[0] = contribution_1;
 
        // Contribution 2
        auto contribution_2 = z_perm_shift * lagrange_last;
        expected_values[1] = contribution_2;
 
        // Contribution 3
        auto contribution_3 = lagrange_first * z_perm;
        expected_values[2] = contribution_3;
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, DeltaRangeConstraintRelation)
{
    const auto run_test = [](bool random_inputs) {
        using Relation = DeltaRangeConstraintRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& q_delta_range = input_elements.q_delta_range();
 
        auto delta_1 = w_2 - w_1;
        auto delta_2 = w_3 - w_2;
        auto delta_3 = w_4 - w_3;
        auto delta_4 = w_1_shift - w_4;
 
        auto contribution_1 = delta_1 * (delta_1 - 1) * (delta_1 - 2) * (delta_1 - 3);
        auto contribution_2 = delta_2 * (delta_2 - 1) * (delta_2 - 2) * (delta_2 - 3);
        auto contribution_3 = delta_3 * (delta_3 - 1) * (delta_3 - 2) * (delta_3 - 3);
        auto contribution_4 = delta_4 * (delta_4 - 1) * (delta_4 - 2) * (delta_4 - 3);
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        expected_values[0] = contribution_1 * q_delta_range;
        expected_values[1] = contribution_2 * q_delta_range;
        expected_values[2] = contribution_3 * q_delta_range;
        expected_values[3] = contribution_4 * q_delta_range;
 
        const auto parameters = RelationParameters<FF>::get_random();
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, EllipticRelation)
{
    const auto run_test = [](bool random_inputs) {
        using Relation = EllipticRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        // Non-const: the test writes `q_l = -1` below to set q_sign explicitly.
        InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
 
        const auto& x_1 = input_elements.w_r();
        const auto& y_1 = input_elements.w_o();
 
        const auto& x_2 = input_elements.w_l_shift();
        const auto& y_2 = input_elements.w_4_shift();
        const auto& x_3 = input_elements.w_r_shift();
        const auto& y_3 = input_elements.w_o_shift();
 
        // In the EllipticRelation, q_l is interpreted as q_sign and can be +1 or -1. Here we explicitly set it to -1
        // (arbitrary, could also be +1) because the relation algebra makes use of the assumption that q_sign^2 = 1.
        // This allows writing the simplified constraint algebra in this test in a more straightforward way.
        input_elements[InputElements::EntityId::q_l] = FF(-1);
        const auto& q_sign = input_elements.q_l();
        const auto& q_elliptic = input_elements.q_elliptic();
        const auto& q_is_double = input_elements.q_m();
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
        // Compute x/y coordinate identities
        {
            auto y_diff = (q_sign * y_2 - y_1);
            auto x_diff = (x_2 - x_1);
            auto x_diff_sqr = x_diff * x_diff;
            auto lambda = y_diff / x_diff;
            auto lambda_sqr = lambda * lambda;
 
            // Contribution (1) point addition, x-coordinate check
            // Formula: x3 = lambda^2 - (x1 + x2)
            // lambda = (y2 - y1) / (x2 - x1)
            // Constraint: (x3 - lambda^2 + (x1 + x2)) * (x2 - x1)^2 = 0
            auto x_add_identity = (x_3 - lambda_sqr + (x_1 + x_2)) * x_diff_sqr;
 
            // Contribution (2) point addition, y-coordinate check
            // Formula: y3 = lambda * (x1 - x3) - y1
            // Constraint: (y3 - lambda * (x1 - x3) + y1) * (x2 - x1) = 0
            auto y_add_identity = (y_3 - lambda * (x_1 - x_3) + y_1) * x_diff;
 
            // N.B. the relation uses the equivalence x1^3 === y1^2 - curve_b to reduce degree by 1 so we must do the
            // same here
            const auto curve_b = EllipticRelationImpl<FF>::get_curve_b();
            auto y1_sqr = (y_1 * y_1);
            auto x_pow_4 = (y1_sqr - curve_b) * x_1; // curve equation substitution
            lambda_sqr = x_pow_4 * 9 / (y1_sqr * 4);
            lambda = (x_1 * x_1 * 3) / (y_1 * 2);
 
            // Contribution (3) point doubling, x-coordinate check
            // Formula: x3 = lambda^2 - 2*x1
            // lambda = (3*x1 * x1) / (2*y1)
            // Constraint: (x3 - lambda^2 + 2*x1) * (2*y1) = 0
            auto x_double_identity = (x_3 - lambda_sqr + x_1 * 2) * (y1_sqr * 4);
 
            // Contribution (4) point doubling, y-coordinate check
            // Formula: y3 = lambda * (x1 - x3) - y1
            // Constraint: (y3 - lambda * (x1 - x3) + y1) * (2*y1) * FF(-1) = 0
            // N.B. multiply by -1 to match form used for efficient accumulation in relation
            auto y_double_identity = (y_3 - lambda * (x_1 - x_3) + y_1) * (y_1 * 2) * FF(-1);
 
            // Combine addition and doubling subidentities, each scaled by q_is_double
            expected_values[0] = (x_add_identity * (-q_is_double + 1) + (x_double_identity * q_is_double)) * q_elliptic;
            expected_values[1] = (y_add_identity * (-q_is_double + 1) + (y_double_identity * q_is_double)) * q_elliptic;
        }
 
        const auto parameters = RelationParameters<FF>::get_random();
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, NonNativeFieldRelation)
{
    const auto run_test = [](bool random_inputs) {
        using Relation = NonNativeFieldRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& w_2_shift = input_elements.w_r_shift();
        const auto& w_3_shift = input_elements.w_o_shift();
        const auto& w_4_shift = input_elements.w_4_shift();
 
        const auto& q_2 = input_elements.q_r();
        const auto& q_3 = input_elements.q_o();
        const auto& q_4 = input_elements.q_4();
        const auto& q_m = input_elements.q_m();
        const auto& q_nnf = input_elements.q_nnf();
 
        constexpr FF LIMB_SIZE(uint256_t(1) << 68);
        constexpr FF SUBLIMB_SHIFT(uint256_t(1) << 14);
        constexpr FF SUBLIMB_SHIFT_2(SUBLIMB_SHIFT * SUBLIMB_SHIFT);
        constexpr FF SUBLIMB_SHIFT_3(SUBLIMB_SHIFT_2 * SUBLIMB_SHIFT);
        constexpr FF SUBLIMB_SHIFT_4(SUBLIMB_SHIFT_3 * SUBLIMB_SHIFT);
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        // [(w_1 * w_2_shift) + (w_1_shift * w_2)] * LIMB_SIZE + (w_1_shift * w_2_shift) - (w_3 + w_4)
        auto nnf_gate_1 = (w_1 * w_2_shift + w_1_shift * w_2) * LIMB_SIZE;
        nnf_gate_1 += (w_1_shift * w_2_shift);
        nnf_gate_1 -= (w_3 + w_4);
 
        // [(w_1 * w_4) + (w_2 * w_3) - w_3_shift] * LIMB_SIZE - w_4_shift + (w_1 * w_2_shift) + (w_1_shift * w_2)
        auto nnf_gate_2 = (w_1 * w_4 + w_2 * w_3 - w_3_shift) * LIMB_SIZE;
        nnf_gate_2 -= w_4_shift;
        nnf_gate_2 += w_1 * w_2_shift + w_1_shift * w_2;
 
        // [(w_1 * w_2_shift) + (w_1_shift * w_2)] * LIMB_SIZE + (w_1_shift * w_2_shift) + w_4 - (w_3_shift + w_4_shift)
        auto nnf_gate_3 = (w_1 * w_2_shift + w_1_shift * w_2) * LIMB_SIZE;
        nnf_gate_3 += (w_1_shift * w_2_shift);
        nnf_gate_3 += w_4;
        nnf_gate_3 -= (w_3_shift + w_4_shift);
 
        auto limb_accumulator_1 = w_1 + w_2 * SUBLIMB_SHIFT + w_3 * SUBLIMB_SHIFT_2 + w_1_shift * SUBLIMB_SHIFT_3 +
                                  w_2_shift * SUBLIMB_SHIFT_4 - w_4;
 
        auto limb_accumulator_2 = w_3 + w_4 * SUBLIMB_SHIFT + w_1_shift * SUBLIMB_SHIFT_2 +
                                  w_2_shift * SUBLIMB_SHIFT_3 + w_3_shift * SUBLIMB_SHIFT_4 - w_4_shift;
 
        // Multiply each subidentity by its corresponding selector product
        nnf_gate_1 *= (q_2 * q_3);
        nnf_gate_2 *= (q_2 * q_4);
        nnf_gate_3 *= (q_2 * q_m);
        limb_accumulator_1 *= (q_3 * q_4);
        limb_accumulator_2 *= (q_3 * q_m);
 
        auto non_native_field_identity = nnf_gate_1 + nnf_gate_2 + nnf_gate_3;
        auto limb_accumulator_identity = limb_accumulator_1 + limb_accumulator_2;
 
        expected_values[0] = non_native_field_identity + limb_accumulator_identity;
        expected_values[0] *= q_nnf;
 
        const auto parameters = RelationParameters<FF>::get_random();
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, MemoryRelation)
{
    const auto run_test = [](bool random_inputs) {
        using Relation = MemoryRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
 
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& w_2_shift = input_elements.w_r_shift();
        const auto& w_3_shift = input_elements.w_o_shift();
        const auto& w_4_shift = input_elements.w_4_shift();
 
        const auto& q_1 = input_elements.q_l();
        const auto& q_2 = input_elements.q_r();
        const auto& q_3 = input_elements.q_o();
        const auto& q_4 = input_elements.q_4();
        const auto& q_m = input_elements.q_m();
        const auto& q_c = input_elements.q_c();
        const auto& q_memory = input_elements.q_memory();
 
        const auto parameters = RelationParameters<FF>::get_random();
        const auto& eta = parameters.eta;
        const auto& eta_two = parameters.eta_two;
        const auto& eta_three = parameters.eta_three;
        const auto& rom_logup_gamma = parameters.rom_logup_gamma;
 
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        auto memory_record_check = w_3 * eta_three;
        memory_record_check += w_2 * eta_two;
        memory_record_check += w_1 * eta;
        memory_record_check += q_c;
        auto partial_record_check = memory_record_check; // used in RAM consistency check
        memory_record_check = memory_record_check - w_4;
 
        auto index_delta = w_1_shift - w_1;
        auto record_delta = w_4_shift - w_4;
 
        auto index_is_monotonically_increasing = index_delta * index_delta - index_delta;
 
        // auto adjacent_values_match_if_adjacent_indices_match = (FF(1) - index_delta) * record_delta;
        auto adjacent_values_match_if_adjacent_indices_match = (index_delta * FF(-1) + FF(1)) * record_delta;
 
        expected_values[1] = adjacent_values_match_if_adjacent_indices_match * (q_1 * q_2);
        expected_values[2] = index_is_monotonically_increasing * (q_1 * q_2);
        auto ROM_consistency_check_identity = memory_record_check * (q_1 * q_2);
 
        auto access_type = (w_4 - partial_record_check);             // will be 0 or 1 for honest Prover
        auto access_check = access_type * access_type - access_type; // check value is 0 or 1
 
        auto next_gate_access_type = w_3_shift * eta_three;
        next_gate_access_type += w_2_shift * eta_two;
        next_gate_access_type += w_1_shift * eta;
        next_gate_access_type = w_4_shift - next_gate_access_type;
 
        auto value_delta = w_3_shift - w_3;
        auto adjacent_values_match_if_adjacent_indices_match_and_next_access_is_a_read_operation =
            (index_delta * FF(-1) + FF(1)) * value_delta * (next_gate_access_type * FF(-1) + FF(1));
 
        // We can't apply the RAM consistency check identity on the final entry in the sorted list (the wires in the
        // next gate would make the identity fail). We need to validate that its 'access type' bool is correct. Can't do
        // with an arithmetic gate because of the `eta` factors. We need to check that the *next* gate's access type is
        // correct, to cover this edge case
        auto next_gate_access_type_is_boolean = next_gate_access_type * next_gate_access_type - next_gate_access_type;
 
        // Putting it all together...
        expected_values[3] =
            adjacent_values_match_if_adjacent_indices_match_and_next_access_is_a_read_operation * (q_3);
        expected_values[4] = index_is_monotonically_increasing * (q_3);
        expected_values[5] = next_gate_access_type_is_boolean * (q_3);
        auto RAM_consistency_check_identity = access_check * (q_3);
 
        memory_record_check *= (q_1 * q_m);
 
        auto timestamp_delta = w_2_shift - w_2;
        auto RAM_timestamp_check_identity = (index_delta * FF(-1) + FF(1)) * timestamp_delta - w_3;
        RAM_timestamp_check_identity *= (q_1 * q_4);
 
        auto memory_identity = ROM_consistency_check_identity;
        memory_identity += RAM_timestamp_check_identity;
        memory_identity += memory_record_check;
        memory_identity += RAM_consistency_check_identity;
 
        expected_values[0] = memory_identity;
        expected_values[0] *= q_memory;
        expected_values[1] *= q_memory;
        expected_values[2] *= q_memory;
        expected_values[3] *= q_memory;
        expected_values[4] *= q_memory;
        expected_values[5] *= q_memory;
 
        auto q_logup_table = q_2 * (q_1 * FF(-1) + FF(1));
        auto q_logup_read = q_4 * (q_1 * FF(-1) + FF(1));
        auto denom = rom_logup_gamma + w_1 + w_2 * eta + q_c * eta_two;
        // Subrelation 6: inverse correctness (per-row).
        expected_values[6] = q_memory * (q_logup_table + q_logup_read) * (w_4 * denom - FF(1));
        // Subrelation 7: LogUp sum (linearly dependent, no scaling factor).
        expected_values[7] = q_memory * (q_logup_read - q_logup_table * w_3) * w_4;
 
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, Poseidon2ExternalRelation)
{
    const auto run_test = []([[maybe_unused]] bool random_inputs) {
        using Relation = Poseidon2ExternalRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
 
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& w_2_shift = input_elements.w_r_shift();
        const auto& w_3_shift = input_elements.w_o_shift();
        const auto& w_4_shift = input_elements.w_4_shift();
        const auto& q_1 = input_elements.q_l();
        const auto& q_2 = input_elements.q_r();
        const auto& q_3 = input_elements.q_o();
        const auto& q_4 = input_elements.q_4();
        const auto& q_poseidon2_external = input_elements.q_poseidon2_external();
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        // add round constants
        auto s1 = w_1 + q_1;
        auto s2 = w_2 + q_2;
        auto s3 = w_3 + q_3;
        auto s4 = w_4 + q_4;
 
        // apply s-box round
        auto u1 = s1 * s1;
        u1 *= u1;
        u1 *= s1;
        auto u2 = s2 * s2;
        u2 *= u2;
        u2 *= s2;
        auto u3 = s3 * s3;
        u3 *= u3;
        u3 *= s3;
        auto u4 = s4 * s4;
        u4 *= u4;
        u4 *= s4;
 
        // matrix mul v = M_E * u with 14 additions
        auto t0 = u1 + u2; // u_1 + u_2
        auto t1 = u3 + u4; // u_3 + u_4
        auto t2 = u2 + u2; // 2u_2
        t2 += t1;          // 2u_2 + u_3 + u_4
        auto t3 = u4 + u4; // 2u_4
        t3 += t0;          // u_1 + u_2 + 2u_4
        auto v4 = t1 + t1;
        v4 += v4;
        v4 += t3; // u_1 + u_2 + 4u_3 + 6u_4
        auto v2 = t0 + t0;
        v2 += v2;
        v2 += t2;          // 4u_1 + 6u_2 + u_3 + u_4
        auto v1 = t3 + v2; // 5u_1 + 7u_2 + u_3 + 3u_4
        auto v3 = t2 + v4; // u_1 + 3u_2 + 5u_3 + 7u_4
 
        // output is { v1, v2, v3, v4 }
 
        expected_values[0] = q_poseidon2_external * (v1 - w_1_shift);
        expected_values[1] = q_poseidon2_external * (v2 - w_2_shift);
        expected_values[2] = q_poseidon2_external * (v3 - w_3_shift);
        expected_values[3] = q_poseidon2_external * (v4 - w_4_shift);
 
        const auto parameters = RelationParameters<FF>::get_random();
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
 
        // validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};
 
TEST_F(UltraRelationConsistency, Poseidon2InternalRelation)
{
    const auto run_test = []([[maybe_unused]] bool random_inputs) {
        using Relation = Poseidon2InternalRelation<FF>;
        using SumcheckArrayOfValuesOverSubrelations = typename Relation::SumcheckArrayOfValuesOverSubrelations;
        const InputElements input_elements = random_inputs ? get_random_inputs() : get_special_inputs();
 
        const auto& w_1 = input_elements.w_l();
        const auto& w_2 = input_elements.w_r();
        const auto& w_3 = input_elements.w_o();
        const auto& w_4 = input_elements.w_4();
        const auto& w_1_shift = input_elements.w_l_shift();
        const auto& w_2_shift = input_elements.w_r_shift();
        const auto& w_3_shift = input_elements.w_o_shift();
        const auto& w_4_shift = input_elements.w_4_shift();
        const auto& q_1 = input_elements.q_l();
        const auto& q_poseidon2_internal = input_elements.q_poseidon2_internal();
        SumcheckArrayOfValuesOverSubrelations expected_values;
 
        // add round constants on only first element
        auto v1 = w_1 + q_1;
 
        // apply s-box to only first element
        auto u1 = v1 * v1;
        u1 *= u1;
        u1 *= v1;
 
        // multiply with internal matrix
        // Uses D_i - 1 values: result[i] = (D_i - 1) * x[i] + sum = D_i * x[i] + (sum of other elements)
        auto sum = u1 + w_2 + w_3 + w_4;
        auto t0 = u1 * crypto::Poseidon2Bn254ScalarFieldParams::internal_matrix_diagonal_minus_one[0];
        t0 += sum;
        auto t1 = w_2 * crypto::Poseidon2Bn254ScalarFieldParams::internal_matrix_diagonal_minus_one[1];
        t1 += sum;
        auto t2 = w_3 * crypto::Poseidon2Bn254ScalarFieldParams::internal_matrix_diagonal_minus_one[2];
        t2 += sum;
        auto t3 = w_4 * crypto::Poseidon2Bn254ScalarFieldParams::internal_matrix_diagonal_minus_one[3];
        t3 += sum;
 
        expected_values[0] = q_poseidon2_internal * (t0 - w_1_shift);
        expected_values[1] = q_poseidon2_internal * (t1 - w_2_shift);
        expected_values[2] = q_poseidon2_internal * (t2 - w_3_shift);
        expected_values[3] = q_poseidon2_internal * (t3 - w_4_shift);
 
        const auto parameters = RelationParameters<FF>::get_random();
        validate_relation_execution<Relation>(expected_values, input_elements, parameters);
 
        // validate_relation_execution<Relation>(expected_values, input_elements, parameters);
    };
    run_test(/*random_inputs=*/false);
    run_test(/*random_inputs=*/true);
};