def testVerbosity(self):
    tensorboard_logging.set_summary_writer(self._sw)
    tensorboard_logging.set_verbosity(tensorboard_logging.ERROR)
    tensorboard_logging.warn("warn")
    tensorboard_logging.error("error")
    tensorboard_logging.set_verbosity(tensorboard_logging.DEBUG)
    tensorboard_logging.debug("debug")

    self.assertLoggedMessagesAre([(event_pb2.LogMessage.ERROR, "error"),
                                  (event_pb2.LogMessage.DEBUGGING, "debug")])
    # All message should be logged because tensorboard_logging verbosity doesn't
    # affect logging verbosity.
    self.assertEqual(3, self.logged_message_count)
Exemplo n.º 2
0
    def testVerbosity(self):
        tensorboard_logging.set_summary_writer(self._sw)
        tensorboard_logging.set_verbosity(tensorboard_logging.ERROR)
        tensorboard_logging.warn("warn")
        tensorboard_logging.error("error")
        tensorboard_logging.set_verbosity(tensorboard_logging.DEBUG)
        tensorboard_logging.debug("debug")

        self.assertLoggedMessagesAre([(event_pb2.LogMessage.ERROR, "error"),
                                      (event_pb2.LogMessage.DEBUG, "debug")])
        # All message should be logged because tensorboard_logging verbosity doesn't
        # affect logging verbosity.
        self.assertEqual(3, self.logged_message_count)
  def testBadVerbosity(self):
    with self.assertRaises(ValueError):
      tensorboard_logging.set_verbosity("failure")

    with self.assertRaises(ValueError):
      tensorboard_logging.log("bad", "dead")