FIXUP2 TurboSHAKE tests

RustCrypto · Mar 16, 2023 · 0c7db92 · 0c7db92
1 parent 098df34
commit 0c7db92
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sha3/tests/turboshake.rs b/sha3/tests/turboshake.rs
@@ -4,15 +4,13 @@ use digest::ExtendableOutput;
 pub(crate) fn turbo_shake_test<D, F>(
     input: &[u8],
     output: &[u8],
-    truncate_output: u64,
+    truncate_output: usize,
     new: F,
 ) -> Option<&'static str>
 where
     D: ExtendableOutput + Debug + Clone,
     F: Fn() -> D,
 {
-    let truncate_output: usize = truncate_output.try_into().unwrap();
-
     let mut hasher = new();
     let mut buf = [0u8; 16 * 1024];
     let buf = &mut buf[..truncate_output + output.len()];
@@ -76,7 +74,7 @@ macro_rules! new_turbo_shake_test {
                 if let Some(desc) = $test_func(
                     &input,
                     output,
-                    u64::from_be_bytes(truncate_output.try_into().unwrap()),
+                    u64::from_be_bytes(truncate_output.try_into().unwrap()).try_into().unwrap(),
                     || <$hasher>::from_core(<$hasher_core>::new(domain_separation[0])),
                 ) {
                     panic!(