Simplify visit unary op (#408)

sourcefrog · Sep 12, 2024 · 727682a · 727682a
2 parents da63492 + 9b4cc9d
commit 727682a
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/visit.rs b/src/visit.rs
@@ -504,20 +504,17 @@ impl<'ast> Visit<'ast> for DiscoveryVisitor<'_> {
         if attrs_excluded(&i.attrs) {
             return;
         }
-        let replacements = match i.op {
-            UnOp::Not(_) => vec![quote! {}],
-            UnOp::Neg(_) => vec![quote! {}],
+        match i.op {
+            UnOp::Not(_) | UnOp::Neg(_) => {
+                self.collect_mutant(i.op.span().into(), quote! {}, Genre::UnaryOperator);
+            }
             _ => {
                 trace!(
                     op = i.op.to_pretty_string(),
                     "No mutants generated for this unary operator"
                 );
-                Vec::new()
             }
         };
-        replacements
-            .into_iter()
-            .for_each(|rep| self.collect_mutant(i.op.span().into(), rep, Genre::UnaryOperator));
         syn::visit::visit_expr_unary(self, i);
     }
 }