async def test_delayed_fail(conf, game):
"""Test failing in a little bit of time"""
prof = game.random_profile()
cmd = ["bash", "-c", "sleep 1 && false"]
async with simsched.simsched(game, conf, cmd) as sched:
with pytest.raises(RuntimeError):
await sched.sample_payoffs(prof)
async def test_read_delay_fail(conf, game):
"""Test script that fails after reading some input"""
cmd = ["bash", "-c", "read line && sleep 0.1 && false"]
async with simsched.simsched(game, conf, cmd) as sched:
future = asyncio.ensure_future(
sched.sample_payoffs(game.random_profile()))
with pytest.raises(RuntimeError):
await future
async def test_immediate_fail(conf, game):
"""Test failing immediately"""
prof = game.random_profile()
cmd = ["bash", "-c", "false"]
async with simsched.simsched(game, conf, cmd) as sched:
with pytest.raises(RuntimeError):
await asyncio.gather(sched.sample_payoffs(prof),
sched.sample_payoffs(prof))
async def test_early_exit(conf, game):
"""Test closing while some futures are unresolved"""
profs = game.random_profiles(20)
cmd = ["bash", "-c", "while read line; do : ; done"]
async with simsched.simsched(game, conf, cmd) as sched:
# Schedule promises but don't await
asyncio.ensure_future(
asyncio.gather(*[sched.sample_payoffs(p) for p in profs]))
await asyncio.sleep(1)
async def test_json_decode_fail(conf, game):
"""Test exist if json is not valid"""
cmd = [
'bash', '-c', 'read line && echo "[" && while read line; do :; done'
]
prof = game.random_profile()
async with simsched.simsched(game, conf, cmd) as sched:
with pytest.raises(json.decoder.JSONDecodeError):
await asyncio.gather(sched.sample_payoffs(prof),
sched.sample_payoffs(prof),
sched.sample_payoffs(prof))
async def test_buffer_blocking(conf, game):
"""Test that buffer blocking works if we try to write a lot of data"""
prof = game.random_profile()
cmd = ["python3", "cdasim/sim.py", "--single", "1"]
async with simsched.simsched(game, conf, cmd, buff_size=4096) as sched:
await sched.sample_payoffs(prof)
bprof = json.dumps(sched._base, separators=(",", ":")).encode("utf8") # pylint: disable=protected-access
# number we have to write to blow out buffer
num = 5 * math.ceil(4096 / (len(bprof) + 1))
await asyncio.gather(
*[sched.sample_payoffs(p) for p in itertools.repeat(prof, num)])
async def test_read_delay_schedule_fail(conf, game):
"""Test script that fails after reading some input"""
cmd = ["bash", "-c", "read line && sleep 0.2 && false"]
async with simsched.simsched(game, conf, cmd) as sched:
future = asyncio.ensure_future(
sched.sample_payoffs(game.random_profile()))
await asyncio.sleep(1) # make sure process is dead
# Since we're using asyncio, the future actually gets assigned the
# error before this one kicks off.
asyncio.ensure_future(sched.sample_payoffs(game.random_profile()))
with pytest.raises(RuntimeError):
await future
async def test_basic_profile(conf, game):
"""Test sampling a basic profile"""
profs = game.random_profiles(20)
cmd = ["python3", "cdasim/sim.py", "--single", "1"]
async with simsched.simsched(game, conf, cmd) as sched:
assert str(sched) == "python3 cdasim/sim.py --single 1"
assert rsgame.empty_copy(sched) == game
awaited = await asyncio.gather(
*[sched.sample_payoffs(p) for p in profs])
pays = np.stack(awaited)
assert pays.shape == profs.shape
assert np.allclose(pays[profs == 0], 0)
assert np.any(pays != 0)
async def create_scheduler(game, command, conf=None, buff=65536, count=1):
"""Create a simulation scheduler"""
assert int(count) > 0
buff_size = int(buff)
if game == "-":
rsgame = gamereader.load(sys.stdin)
else:
with open(game) as fil:
rsgame = gamereader.load(fil)
if conf is None:
config = {}
elif conf == "-":
config = json.load(sys.stdin)
else:
with open(conf) as fil:
config = json.load(fil)
return simsched.simsched(rsgame, config, command.split(), buff_size=buff_size)
async def test_bad_command_exit(conf, game):
"""Test proper exit if command doesn't exist"""
with pytest.raises(FileNotFoundError):
async with simsched.simsched(game, conf, ["unknown"]):
pass # pragma: no cover never called
async def test_ignore_terminate_fail(conf, game):
"""Test exit if script stops"""
cmd = ["bash", "-c", 'trap "" SIGTERM && sleep 20']
async with simsched.simsched(game, conf, cmd):
# Wait for term to be captured
await asyncio.sleep(1)
