use winter_air::Air;
use winter_math::fields::f64::BaseElement as Felt;
use winterfell::{Trace, TraceTable};
use crate::{
evaluators::evaluators::{EvaluatorsAir, PublicInputs},
helpers::{AirTester, MyTraceTable},
};
#[derive(Clone)]
struct EvaluatorsAirTester {}
impl AirTester for EvaluatorsAirTester {
type PubInputs = PublicInputs;
fn build_main_trace(&self, length: usize) -> MyTraceTable {
let trace_width = 7;
let mut trace = TraceTable::new(trace_width, length);
let start = Felt::new(0);
trace.fill(
|state| {
state[0] = start;
state[1] = start;
state[3] = start;
state[4] = start;
state[5] = start;
state[6] = start;
},
|_, state| {},
);
MyTraceTable::new(trace, 0)
}
fn public_inputs(&self) -> PublicInputs {
let zero = Felt::new(0);
PublicInputs::new([zero; 16])
}
}
#[test]
fn test_evaluators_air() {
let air_tester = Box::new(EvaluatorsAirTester {});
let length = 1024;
let main_trace = air_tester.build_main_trace(length);
let aux_trace = air_tester.build_aux_trace(length);
let pub_inputs = air_tester.public_inputs();
let trace_info = air_tester.build_trace_info(length);
let options = air_tester.build_proof_options();
let air = EvaluatorsAir::new(trace_info, pub_inputs, options);
main_trace.validate::<EvaluatorsAir, Felt>(&air, aux_trace.as_ref());
}