def METHOD_NAME(assertion):
    start = datetime.now()
    while (datetime.now() - start).seconds < 120:
        sleep(0.1)  # Limit CPU usage
        if assertion():
            return
    assert assertion()