try to fix 'RuntimeError: No active exception to reraise'

1 files changed, 4 insertions, 3 deletions

diff --git a/test/test_psutil.py b/test/test_psutil.py
index 1bff2064..5d37fc32 100644
--- a/test/test_psutil.py
+++ b/test/test_psutil.py
@@ -460,12 +460,13 @@ def retry_before_failing(ntimes=None):
         def wrapper(*args, **kwargs):
             times = ntimes or NO_RETRIES
             assert times, times
+            exc = None
             for x in range(times):
                 try:
                     return fun(*args, **kwargs)
-                except AssertionError:
-                    pass
-            raise
+                except AssertionError as _:
+                    exc = _
+            raise exc
         return wrapper
     return decorator