async def test_basic_profile(game):
"""Test scheduling a standard profile"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(
server.create_simulator("sim",
"1",
delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(
zip(game.role_names, game.num_role_players, game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim["id"], symgrps)
profs = game.random_profiles(20)
# Schedule all new profiles and verify it works
async with eosched.eosched(game, egta, egame["id"], 0.1, 1, 10, 0,
0) as sched:
assert str(sched) == str(egame["id"])
assert game == rsgame.empty_copy(sched)
awaited = await asyncio.gather(
*[sched.sample_payoffs(p) for p in profs])
pays = np.stack(awaited)
assert np.allclose(pays[profs == 0], 0)
# Schedule old profiles and verify it still works
async with eosched.eosched(game, egta, egame["id"], 0.1, 1, 10, 0,
0) as sched:
awaited = await asyncio.gather(
*[sched.sample_payoffs(p) for p in profs])
pays = np.stack(awaited)
assert np.allclose(pays[profs == 0], 0)
# Schedule two at a time, in two batches
async with eosched.eosched(game, egta, egame["id"], 0.1, 2, 10, 0,
0) as base_sched:
sched = countsched.countsched(base_sched, 2)
awaited = await asyncio.gather(
*[sched.sample_payoffs(p) for p in profs])
pays = np.stack(awaited)
assert np.allclose(pays[profs == 0], 0)
# Try again now that everything should be scheduled
async with eosched.eosched(game, egta, egame["id"], 0.1, 2, 10, 0,
0) as base_sched:
sched = countsched.countsched(base_sched, 2)
awaited = await asyncio.gather(
*[sched.sample_payoffs(p) for p in profs])
pays = np.stack(awaited)
assert np.allclose(pays[profs == 0], 0)
async def test_exception_in_get(game):
"""Test exception in await"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(
server.create_simulator("sim",
"1",
delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(
zip(game.role_names, game.num_role_players, game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim["id"], symgrps)
profs = game.random_profiles(20)
async with eosched.eosched(game, egta, egame["id"], 0.1, 1, 10, 0,
0) as sched:
tasks = [
asyncio.ensure_future(sched.sample_payoffs(p)) for p in profs
]
await asyncio.sleep(0.1)
server.custom_response(lambda: _raise(TimeoutError))
await asyncio.sleep(0.1)
with pytest.raises(TimeoutError):
await asyncio.gather(*tasks)
# tidy up
errors = asyncio.gather(*tasks, return_exceptions=True)
errors.cancel()
with contextlib.suppress(TimeoutError, asyncio.CancelledError):
await errors
async def test_exception_in_create(game):
"""Test exception creating eo scheduler"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(server.create_simulator( # pragma: no branch pylint: disable=line-too-long
'sim', '1', delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(zip(game.role_names, game.num_role_players,
game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim['id'], symgrps)
server.custom_response(lambda: _raise(TimeoutError))
with pytest.raises(TimeoutError):
async with eosched.eosched(
game, egta, egame['id'], 0.1, 1, 25, 0, 0):
pass # pragma: no cover
async def test_scheduler_deactivate(game):
"""Test that scheduler ends when deactivated"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(server.create_simulator( # pragma: no branch pylint: disable=line-too-long
'sim', '1', delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(zip(game.role_names, game.num_role_players,
game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim['id'], symgrps)
# Schedule all new profiles and verify it works
# This first time should have to wait to schedule more
async with eosched.eosched(
game, egta, egame['id'], 0.1, 1, 10, 0, 0) as sched:
# Deactivate scheduler
for esched in await egta.get_generic_schedulers():
await esched.deactivate()
with pytest.raises(ValueError):
await sched.sample_payoffs(game.random_profile())
async def test_exception_in_schedule(game):
"""Test exception throwin in scheduler"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(server.create_simulator( # pragma: no branch pylint: disable=line-too-long
'sim', '1', delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(zip(game.role_names, game.num_role_players,
game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim['id'], symgrps)
prof = game.random_profile()
async with eosched.eosched(
game, egta, egame['id'], 0.1, 1, 25, 0, 0) as sched:
# so that enough calls to get_requirements are made
server.custom_response(lambda: _raise(TimeoutError))
await asyncio.sleep(0.1)
with pytest.raises(TimeoutError):
await sched.sample_payoffs(prof)
async def test_exception_in_get(game):
"""Test exception in await"""
async with mockserver.server() as server, api.api() as egta:
sim = await egta.get_simulator(server.create_simulator(
'sim', '1', delay_dist=lambda: random.random() / 10))
strats = dict(zip(game.role_names, game.strat_names))
symgrps = list(zip(game.role_names, game.num_role_players,
game.strat_names))
await sim.add_strategies(strats)
egame = await egta.get_canon_game(sim['id'], symgrps)
profs = game.random_profiles(20)
async with eosched.eosched(
game, egta, egame['id'], 0.1, 1, 10, 0, 0) as sched:
futures = asyncio.gather(*[
sched.sample_payoffs(p) for p in profs])
await asyncio.sleep(0.1)
server.custom_response(lambda: _raise(TimeoutError))
await asyncio.sleep(0.1)
with pytest.raises(TimeoutError):
await futures