iterator

tests/test_postgresql.py

@@ -65,13 +65,15 @@ def wait_for(func, timeout=60):
     """Check for function to finish."""
     time: datetime = datetime.utcnow()
     timeout_diff: timedelta = timedelta(seconds=timeout)
+    i = 0
     while True:
+        i += 1
         try:
             func()
             return
         except AssertionError as e:
             if time + timeout_diff < datetime.utcnow():
-                raise e
+                raise AssertionError("Faile after {i} attempts".format(i=i)) from e
             sleep(1)
             pass