def __init__(self, player_config, env_config):
player_base.PlayerBase.__init__(self, player_config)
self._can_play_right = True
self._replay = script_helpers.ScriptHelpers().load_dump(
player_config['path'])
self._step = 0
self._player = player_config['index']
def compute_hash(self, trace_file):
replay = script_helpers.ScriptHelpers().load_dump(trace_file)
hash_value = 0
for frame in replay:
del frame['debug']
hash_value = zlib.adler32(
str(tuple(sorted(frame.items()))).encode(), hash_value)
return hash_value
def test__replay(self):
"""Has to run first, as it generates dumps for other tests."""
dumps_before = self.current_dumps()
self.generate_replay()
dumps_after = self.current_dumps()
dump1 = dumps_after - dumps_before
assert len(dump1) == 1, dump1
hash1 = self.compute_hash(list(dump1)[0])
cfg = {
'dump_full_episodes': True,
}
script_helpers.ScriptHelpers().replay(list(dump1)[0],
directory=test_tmpdir,
config_update=cfg,
render=False)
dump2 = self.current_dumps() - dumps_after
assert len(dump2) == 1, dump2
hash2 = self.compute_hash(list(dump2)[0])
self.assertEqual(hash1, hash2)
def main(_): script_helpers.ScriptHelpers().replay(FLAGS.trace_file, FLAGS.fps)
def main(_):
trace_file = '/home/mjbigdel/Downloads/episode_done_20201214-191941506361.dump'
fps = 10
# script_helpers.ScriptHelpers().replay(FLAGS.trace_file, FLAGS.fps)
script_helpers.ScriptHelpers().replay(trace_file, fps)
def test_dump_to_txt(self):
dump = list(self.current_dumps())[0]
out_file = dump.replace('dump', 'txt')
script_helpers.ScriptHelpers().dump_to_txt(dump, out_file, True)
self.assertTrue(os.path.isfile(out_file))
def test_dump_to_video(self):
dump = list(self.current_dumps())[0]
script_helpers.ScriptHelpers().dump_to_video(dump)
def main(_):
script_helpers.ScriptHelpers().dump_to_txt(FLAGS.trace_file, FLAGS.output,
FLAGS.include_debug)
def main(_):
script_helpers.ScriptHelpers().dump_to_video(FLAGS.trace_file)
