python/ray/_private/test_utils.py
def wait_until_succeeded_without_exception(
    func, exceptions, *args, timeout_ms=1000, retry_interval_ms=100, raise_last_ex=False
):
    """A helper function that waits until a given function
        completes without exceptions.

    Args:
        func: A function to run.
        exceptions(tuple): Exceptions that are supposed to occur.
        args: arguments to pass for a given func
        timeout_ms: Maximum timeout in milliseconds.
        retry_interval_ms: Retry interval in milliseconds.
        raise_last_ex: Raise the last exception when timeout.

    Return:
        Whether exception occurs within a timeout.
    """
    if type(exceptions) != tuple:
        raise Exception("exceptions arguments should be given as a tuple")

    time_elapsed = 0
    start = time.time()
    last_ex = None
    while time_elapsed <= timeout_ms:
        try:
            func(*args)
            return True
        except exceptions as ex:
            last_ex = ex
            time_elapsed = (time.time() - start) * 1000
            time.sleep(retry_interval_ms / 1000.0)
    if raise_last_ex:
        ex_stack = (
            traceback.format_exception(type(last_ex), last_ex, last_ex.__traceback__)
            if last_ex
            else []
        )
        ex_stack = "".join(ex_stack)
        raise Exception(f"Timed out while testing, {ex_stack}")
    return False
A helper function that waits until a given function
        completes without exceptions.

    Args:
        func: A function to run.
        exceptions(tuple): Exceptions that are supposed to occur.
        args: arguments to pass for a given func
        timeout_ms: Maximum timeout in milliseconds.
        retry_interval_ms: Retry interval in milliseconds.
        raise_last_ex: Raise the last exception when timeout.

    Return:
        Whether exception occurs within a timeout.
    
A helper function that waits until a given function
completes without exceptions.
A helper function that waits until a given function completes without exceptions .
def wait_until_succeeded_without_exception(
    func, exceptions, *args, timeout_ms=1000, retry_interval_ms=100, raise_last_ex=False
):
    if type(exceptions) != tuple:
        raise Exception("exceptions arguments should be given as a tuple")
    time_elapsed = 0
    start = time.time()
    last_ex = None
    while time_elapsed <= timeout_ms:
        try:
            func(*args)
            return True
        except exceptions as ex:
            last_ex = ex
            time_elapsed = (time.time() - start) * 1000
            time.sleep(retry_interval_ms / 1000.0)
    if raise_last_ex:
        ex_stack = (
            traceback.format_exception(type(last_ex), last_ex, last_ex.__traceback__)
            if last_ex
            else []
        )
        ex_stack = "".join(ex_stack)
        raise Exception(f"Timed out while testing, {ex_stack}")
    return False
def wait_until_succeeded_without_exception ( func , exceptions , * args , timeout_ms = 1000 , retry_interval_ms = 100 , raise_last_ex = False ) : if type ( exceptions ) != tuple : raise Exception ( "exceptions arguments should be given as a tuple" ) time_elapsed = 0 start = time . time ( ) last_ex = None while time_elapsed <= timeout_ms : try : func ( * args ) return True except exceptions as ex : last_ex = ex time_elapsed = ( time . time ( ) - start ) * 1000 time . sleep ( retry_interval_ms / 1000.0 ) if raise_last_ex : ex_stack = ( traceback . format_exception ( type ( last_ex ) , last_ex , last_ex . __traceback__ ) if last_ex else [ ] ) ex_stack = "" . join ( ex_stack ) raise Exception ( f"Timed out while testing, {ex_stack}" ) return False
A helper function that waits until a given function
completes without exceptions.
A helper function that waits until a given function completes without exceptions .
"""A helper function that waits until a given function
        completes without exceptions.

    Args:
        func: A function to run.
        exceptions(tuple): Exceptions that are supposed to occur.
        args: arguments to pass for a given func
        timeout_ms: Maximum timeout in milliseconds.
        retry_interval_ms: Retry interval in milliseconds.
        raise_last_ex: Raise the last exception when timeout.

    Return:
        Whether exception occurs within a timeout.
    """
func
A function to run.
A function to run .
exceptions
Exceptions that are supposed to occur.
Exceptions that are supposed to occur .
timeout_ms
Maximum timeout in milliseconds.
Maximum timeout in milliseconds .
retry_interval_ms
Retry interval in milliseconds.
Retry interval in milliseconds .
raise_last_ex
Raise the last exception when timeout.
Raise the last exception when timeout .
args
arguments to pass for a given func
arguments to pass for a given func