def test_save_metrics_skips_non_float_histogram_data(self, data_to_skip,
mock_summary_histogram):
metrics = {'metric_name': data_to_skip}
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
tb_mngr.save_metrics(metrics, 0)
mock_summary_histogram.assert_not_called()
def test_nonscalar_metrics_are_written(self):
summary_dir = os.path.join(self.get_temp_dir(), 'logdir')
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=summary_dir)
tb_mngr.save_metrics(_create_nonscalar_metrics(), 0)
self.assertTrue(tf.io.gfile.exists(summary_dir))
self.assertLen(tf.io.gfile.listdir(summary_dir), 1)
def test_save_metrics_skips_non_scalar_data(self, data_to_skip,
mock_summary_scalar):
metrics = {'metric_name': data_to_skip}
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
tb_mngr.save_metrics(metrics, 0)
expected_calls = [mock.call('round_num', 0, step=0)]
self.assertCountEqual(expected_calls, mock_summary_scalar.call_args_list)
def test_save_metrics_raises_value_error_if_round_num_is_out_of_order(self):
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
tb_mngr.save_metrics(_create_scalar_metrics(), 1)
with self.assertRaises(ValueError):
tb_mngr.save_metrics(_create_scalar_metrics(), 0)
def test_save_metrics_logs_float_histogram_data(self, data,
mock_summary_histogram):
metrics = {'metric_name': data}
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
tb_mngr.save_metrics(metrics, 0)
expected_calls = [
mock.call('metric_name', data, step=0),
]
self.assertCountEqual(expected_calls, mock_summary_histogram.call_args_list)
def test_update_metrics_returns_flat_dict(self):
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
input_data_dict = _create_scalar_metrics()
appended_data_dict = tb_mngr.update_metrics(0, input_data_dict)
self.assertEqual({
'a/b': 1.0,
'a/c': 2.0,
'round_num': 0.0
}, appended_data_dict)
def test_update_metrics_raises_value_error_if_round_num_is_negative(self):
tb_mngr = tensorboard_manager.TensorBoardManager(
summary_dir=self.get_temp_dir())
with self.assertRaises(ValueError):
tb_mngr.update_metrics(-1, _create_scalar_metrics())