fix ola_stark unittests

Sin7Y · Feb 1, 2024 · 806e56e · 806e56e
1 parent d78cf89
commit 806e56e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/circuits/src/builtins/poseidon/poseidon_chunk_stark.rs b/circuits/src/builtins/poseidon/poseidon_chunk_stark.rs
@@ -286,6 +286,7 @@ impl<F: RichField + Extendable<D>, const D: usize> Stark<F, D> for PoseidonChunk
 
 mod test {
     use core::trace::trace::{PoseidonChunkRow, Trace};
+    use core::types::Field;
     use std::path::PathBuf;
 
     use crate::builtins::poseidon::columns::{

diff --git a/circuits/src/builtins/poseidon/poseidon_stark.rs b/circuits/src/builtins/poseidon/poseidon_stark.rs
@@ -181,6 +181,7 @@ pub fn ctl_filter_with_storage<F: Field>() -> Column<F> {
 
 mod test {
     use core::trace::trace::{PoseidonRow, Trace};
+    use core::types::Field;
     use std::path::PathBuf;
 
     use crate::stark::stark::Stark;

diff --git a/circuits/src/stark/ola_stark.rs b/circuits/src/stark/ola_stark.rs
@@ -664,6 +664,7 @@ mod tests {
     use core::types::merkle_tree::{encode_addr, tree_key_default};
     use core::types::{Field, GoldilocksField};
     use core::vm::transaction::init_tx_context_mock;
+    use std::process::exit;
     use executor::load_tx::init_tape;
     use executor::trace::{gen_storage_hash_table, gen_storage_table};
     use executor::{BatchCacheManager, Process};
@@ -925,11 +926,11 @@ mod tests {
             Ok(_) => {}
             Err(e) => {
                 println!("execute err:{:?}", e);
-                return;
+                panic!("execute program failed");
             }
         }
         let hash_roots = gen_storage_hash_table(&mut process, &mut program, &mut db);
-        gen_storage_table(&mut process, &mut program, hash_roots).unwrap();
+        gen_storage_table(&mut process, &mut program, hash_roots.unwrap()).unwrap();
         program.trace.start_end_roots = (start, db.root_hash());
 
         let inputs = GenerationInputs::default();