@@ -21,8 +21,8 @@ use super::wrap_hasher::BLAKE3_HASH_LENGTH;
 pub const NUM_PUBS: usize = 1;
 pub const NUM_U256: usize = 14;
 pub const NUM_HASH: usize = 363;
-const VALIDATING_TAPS: usize = 1;
-const HASHING_TAPS: usize = NUM_HASH;
+pub const VALIDATING_TAPS: usize = 1;
+pub const HASHING_TAPS: usize = NUM_HASH;
 pub const NUM_TAPS: usize = HASHING_TAPS + VALIDATING_TAPS;
 
 pub type PublicInputs = [ark_bn254::Fr; NUM_PUBS];
 
@@ -5,7 +5,7 @@ use crate::bn254::ell_coeffs::AffinePairing;
 use crate::bn254::ell_coeffs::BnAffinePairing;
 use crate::bn254::fp254impl::Fp254Impl;
 use crate::bn254::fq::Fq;
-use crate::chunk::api::{NUM_PUBS, NUM_TAPS};
+use crate::chunk::api::{NUM_HASH, NUM_PUBS, NUM_TAPS, NUM_U256, VALIDATING_TAPS};
 use crate::chunk::elements::ElementType;
 use crate::treepp;
 use ark_bn254::Bn254;
@@ -175,6 +175,10 @@ fn generate_segments_using_mock_proof(vk: Vkey, skip_evaluation: bool) -> Vec<Se
         pubs,
         &mut None,
     );
+    assert_eq!(
+        segments.len(),
+        NUM_PUBS + NUM_U256 + NUM_HASH + VALIDATING_TAPS
+    );
     segments
 }
 
@@ -320,3 +324,17 @@ pub(crate) fn bitcom_scripts_from_segments(
     }
     bitcom_scripts
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::chunk::api_compiletime_utils::generate_segments_using_mock_vk_and_mock_proof;
+
+    #[test]
+    fn test_generate_segments_using_mock_proof() {
+        let segments = generate_segments_using_mock_vk_and_mock_proof();
+        assert_eq!(
+            segments.len(),
+            super::NUM_PUBS + super::NUM_U256 + super::NUM_HASH + super::VALIDATING_TAPS
+        );
+    }
+}