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)