def testSetEntry(self):
exp_logger = logger.Logger('/tmp/dopamine_tests')
self.assertEqual(len(exp_logger.data.keys()), 0)
key = 'key'
val = [1, 2, 3, 4]
exp_logger[key] = val
expected_dictionary = {}
expected_dictionary[key] = val
self.assertEqual(expected_dictionary, exp_logger.data)
# Calling __setitem__ with the same value should overwrite the previous
# value.
val = 'new value'
exp_logger[key] = val
expected_dictionary[key] = val
self.assertEqual(expected_dictionary, exp_logger.data)
def testLogToFileWithValidDirectory(self):
exp_logger = logger.Logger(self._test_subdir)
self.assertTrue(exp_logger.is_logging_enabled())
key = 'key'
val = [1, 2, 3, 4]
exp_logger[key] = val
expected_dictionary = {}
expected_dictionary[key] = val
self.assertEqual(expected_dictionary, exp_logger.data)
iteration_number = 7
exp_logger.log_to_file('log', iteration_number)
log_file = os.path.join(self._test_subdir,
'log_{}'.format(iteration_number))
with tf.gfile.GFile(log_file, 'rb') as f:
contents = f.read()
self.assertEqual(
contents,
pickle.dumps(expected_dictionary,
protocol=pickle.HIGHEST_PROTOCOL))
def testGarbageCollection(self):
exp_logger = logger.Logger(self._test_subdir)
self.assertTrue(exp_logger.is_logging_enabled())
key = 'key'
val = [1, 2, 3, 4]
exp_logger[key] = val
expected_dictionary = {}
expected_dictionary[key] = val
self.assertEqual(expected_dictionary, exp_logger.data)
deleted_log_files = 7
total_log_files = logger.CHECKPOINT_DURATION + deleted_log_files
for iteration_number in range(total_log_files):
exp_logger.log_to_file('log', iteration_number)
for iteration_number in range(total_log_files):
log_file = os.path.join(self._test_subdir,
'log_{}'.format(iteration_number))
if iteration_number < deleted_log_files:
self.assertFalse(tf.gfile.Exists(log_file))
else:
self.assertTrue(tf.gfile.Exists(log_file))
def testLogToFileWithInvalidDirectory(self):
exp_logger = logger.Logger('/does/not/exist')
self.assertFalse(exp_logger.is_logging_enabled())
exp_logger.log_to_file(None, None)
def testLoggingEnabledWithValidDirectory(self):
exp_logger = logger.Logger('/tmp/dopamine_tests')
self.assertTrue(exp_logger.is_logging_enabled())
def testLoggingDisabledWithInvalidDirectory(self):
exp_logger = logger.Logger('/does/not/exist')
self.assertFalse(exp_logger.is_logging_enabled())
def testLoggingDisabledWithEmptyDirectory(self):
exp_logger = logger.Logger('')
self.assertFalse(exp_logger.is_logging_enabled())
def _create_directories(self):
self.checkpoint_dir = os.path.join(self.base_dir, 'checkpoints')
self.logger = logger.Logger(os.path.join(self.base_dir, 'logs'))
def _create_directories(self):
"""Create necessary sub-directories."""
self._checkpoint_dir = os.path.join(self._base_dir, 'checkpoints')
self._logger = logger.Logger(os.path.join(self._base_dir, 'logs'))
