From 9b0f325b5c1d6645f269e95c118a833fb18f6260 Mon Sep 17 00:00:00 2001 From: "F.N. Claessen" Date: Fri, 10 May 2024 11:16:31 +0200 Subject: [PATCH] fix: review suggestion --- aioresponses/core.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/aioresponses/core.py b/aioresponses/core.py index fdc3d6a..1d5f6fd 100644 --- a/aioresponses/core.py +++ b/aioresponses/core.py @@ -469,11 +469,12 @@ async def match( else: return None - if matcher.repeat in (False, 1): - del self._matches[key] - elif matcher.repeat == 2: - matcher.repeat = False - elif isinstance(matcher.repeat, int): + if isinstance(matcher.repeat, bool): + if not matcher.repeat: + del self._matches[key] + else: + if matcher.repeat == 1: + del self._matches[key] matcher.repeat -= 1 if self.is_exception(response_or_exc):