Python set_flaky_mode Exemples

Exemple #1

def test_reconnect_timeout(train_sim):
    set_flaky_mode(True)
    set_fail_duration(20)
    reset_count()
    train_sim._impl._sim_connection._retry_timeout_seconds = 1
    counter = 0
    while train_sim.run():
        if counter == 100:
            # Avoid infinite simulation loops and fail test
            assert False
        counter += 1

Exemple #2

def train_config():
    set_flaky_mode(False)
    set_unauthorized_mode(False)
    return Config([
        __name__,
        '--accesskey=VALUE',
        '--username=alice',
        '--url=http://localhost:8889',
        '--brain=cartpole',
        '--proxy=VALUE',
    ])

Exemple #3

def predict_config():
    set_predict_mode(True)
    set_flaky_mode(False)
    return Config([
        __name__,
        '--accesskey=VALUE',
        '--username=alice',
        '--url=http://localhost:8889',
        '--brain=cartpole',
        '--proxy=VALUE',
        '--predict=4',
    ])

Exemple #4

def test_reconnect(train_sim, monkeypatch):
    set_flaky_mode(True)
    reset_count()

    def _patched_sleep(backoff):
        return backoff
    monkeypatch.setattr(time, 'sleep', _patched_sleep)

    counter = 0
    while train_sim.run():
        if counter == 100:
            break
        counter += 1

Exemple #5

def test_reconnect_user_does_not_want_to_reconnect(train_sim, capsys):
    set_flaky_mode(True)
    reset_count()
    train_sim._impl._sim_connection._retry_timeout_seconds = 0
    counter = 0
    while train_sim.run():
        if counter == 100:
            # Avoid infinite simulation loops and fail test
            assert False
        counter += 1
    out, err = capsys.readouterr()
    assert err.count('Starting Training') == 1
    assert err.count('Websocket connection closed.') == 1

Exemple #6

Fichier : test_event_pump.py Projet : hsiehpc/bonsai-sdk

def test_server_error(train_sim):
    set_flaky_mode(True)
    reset_count()
    train_sim._impl._sim_connection._retry_timeout_seconds = 0
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeStartEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeFinishEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeStartEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), FinishedEvent)

Exemple #7

Fichier : test_event_pump.py Projet : hsiehpc/bonsai-sdk

def test_unable_to_connect_no_operation(train_sim):
    set_flaky_mode(True)
    reset_count()
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeStartEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeFinishEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), EpisodeStartEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), SimulateEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)
    assert isinstance(train_sim.get_next_event(), UnknownEvent)