zeroized expected output for on-the-fly zeroize test

Author: Mojtaba Bisheh Niasar

src/abr_top/uvmf/uvmf_template_output/project_benches/mldsa/tb/sequences/src/ML_KEM_base_sequence.svh

@@ -331,9 +331,10 @@ class ML_KEM_base_sequence extends mldsa_bench_sequence_base;
     end 
   endtask
 
-  task compare_keygen_vectors();
+  task compare_keygen_vectors(bit error_flag = 0);
     // Compare Encapsulation Key (ek)
     foreach (expected_ek[i]) begin
+      if (error_flag) expected_ek[i] = '0;
       if (actual_ek[i] !== expected_ek[i]) begin
         `uvm_error("EK_MISMATCH",
           $sformatf("ek[%0d] mismatch: expected %08h, actual %08h",
@@ -347,6 +348,7 @@ class ML_KEM_base_sequence extends mldsa_bench_sequence_base;
 
     // Compare Decapsulation Key (dk)
     foreach (expected_dk[j]) begin
+      if (error_flag) expected_dk[j] = '0;
       if (actual_dk[j] !== expected_dk[j]) begin
         `uvm_error("DK_MISMATCH",
           $sformatf("dk[%0d] mismatch: expected %08h, actual %08h",

src/abr_top/uvmf/uvmf_template_output/project_benches/mldsa/tb/sequences/src/ML_KEM_randomized_all_sequence.svh

@@ -48,7 +48,7 @@ class ML_KEM_randomized_all_sequence extends ML_KEM_base_sequence;
     end
     read_ek();
     read_dk();
-    compare_keygen_vectors();
+    compare_keygen_vectors(on_the_fly_zeroize);
     zeroize();
     wait_for_done(0, "ready");
   endtask
@@ -68,7 +68,7 @@ class ML_KEM_randomized_all_sequence extends ML_KEM_base_sequence;
     end
     read_ciphertext();
     read_shared_key();
-    compare_encap_vectors();
+    compare_encap_vectors(on_the_fly_zeroize);
     zeroize();
     wait_for_done(0, "ready");
   endtask
@@ -87,7 +87,7 @@ class ML_KEM_randomized_all_sequence extends ML_KEM_base_sequence;
       wait_for_done(1, "valid");
     end
     read_shared_key();
-    compare_decap_vectors();
+    compare_decap_vectors(on_the_fly_zeroize);
     zeroize();
     wait_for_done(0, "ready");
   endtask
@@ -106,7 +106,7 @@ class ML_KEM_randomized_all_sequence extends ML_KEM_base_sequence;
       wait_for_done(1, "valid");
     end
     read_shared_key();
-    compare_decap_vectors();
+    compare_decap_vectors(on_the_fly_zeroize);
     zeroize();
     wait_for_done(0, "ready");
   endtask