def testIntegrationRainbow(self):
"""Test the rainbow agent."""
tf.logging.info('####### Training the Rainbow agent #####')
tf.logging.info('####### Rainbow base_dir: {}'.format(FLAGS.base_dir))
self.quickRainbowFlags()
train.main([])
self.verifyFilesCreated(FLAGS.base_dir)
shutil.rmtree(FLAGS.base_dir)
def testIntegrationDqn(self):
"""Test the DQN agent."""
tf.logging.info('####### Training the DQN agent #####')
tf.logging.info('####### DQN base_dir: {}'.format(FLAGS.base_dir))
self.quickDqnFlags()
train.main([])
self.verifyFilesCreated(FLAGS.base_dir)
shutil.rmtree(FLAGS.base_dir)
def test_atari_init(self):
"""Tests that a DQN agent is initialized."""
train.main([])
shutil.rmtree(FLAGS.base_dir)