refactor(test): use iterator transform instead of explicit loop

TritonVM · Feb 4, 2024 · fc8b9d2 · fc8b9d2
1 parent c8a8f8c
commit fc8b9d2
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/triton-vm/src/proof_item.rs b/triton-vm/src/proof_item.rs
@@ -194,11 +194,9 @@ pub(crate) mod tests {
 
     #[test]
     fn can_loop_over_proof_item_variants() {
-        let mut all_discriminants = HashSet::new();
-        for variant in ProofItemVariant::iter() {
-            all_discriminants.insert(variant.bfield_codec_discriminant());
-        }
-
+        let all_discriminants: HashSet<_> = ProofItemVariant::iter()
+            .map(|variant| variant.bfield_codec_discriminant())
+            .collect();
         assert_eq!(ProofItem::COUNT, all_discriminants.len());
     }