From e4745cbac5d1d70c15e33ed13cb4871df960f92f Mon Sep 17 00:00:00 2001 From: Tyson Smith Date: Thu, 6 Jun 2024 19:15:30 -0700 Subject: [PATCH] [loki] Simplify aggression checks --- loki/loki.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/loki/loki.py b/loki/loki.py index c5c475b6..b811b5da 100644 --- a/loki/loki.py +++ b/loki/loki.py @@ -27,7 +27,7 @@ class Loki: __slots__ = ("aggr", "byte_order") def __init__( - self, aggression: Optional[float] = 0.0, byte_order: Optional[str] = None + self, aggression: float = 0.0, byte_order: Optional[str] = None ) -> None: """ Arguments: @@ -35,11 +35,9 @@ def __init__( byte_order: Used to indicate big or little endian when mutating multiple bytes at once. """ - assert aggression is not None - assert aggression >= 0 - assert aggression <= 1 + assert 0 <= aggression <= 1 assert byte_order is None or byte_order in self.BYTE_ORDERS - self.aggr = min(max(aggression, 0.0), 1.0) + self.aggr = aggression self.byte_order = byte_order @staticmethod @@ -100,7 +98,7 @@ def _fuzz(self, tgt_fp: IO[bytes]) -> None: LOG.debug("cannot fuzz empty file") return # minimum number of mutations should be 1 - max_mutations = max(int(round(length * self.aggr)), 1) + max_mutations = max(round(length * self.aggr), 1) mutations = randint(1, max_mutations) LOG.debug( "%d of a possible %d mutations will be performed", mutations, max_mutations