Ejemplos de GetCheckpointWriteDurations en Python

Ejemplo n.º 1

    def test_checkpoint_add_write_duration(self):
        self.assertEqual(
            self._get_histogram_proto(
                metrics.GetCheckpointWriteDurations(api_label="foo")).num, 0)

        metrics.AddCheckpointWriteDuration(api_label="foo", microseconds=100)
        metrics.AddCheckpointWriteDuration(api_label="foo", microseconds=200)

        self.assertEqual(
            self._get_histogram_proto(
                metrics.GetCheckpointWriteDurations(api_label="foo")).num, 2)
        self.assertEqual(
            self._get_histogram_proto(
                metrics.GetCheckpointWriteDurations(api_label="foo")).min, 100)
        self.assertEqual(
            self._get_histogram_proto(
                metrics.GetCheckpointWriteDurations(api_label="foo")).max, 200)

Ejemplo n.º 2

 def _get_write_histogram_proto(self, api_label):
   proto_bytes = metrics.GetCheckpointWriteDurations(api_label=api_label)
   histogram_proto = summary_pb2.HistogramProto()
   histogram_proto.ParseFromString(proto_bytes)
   return histogram_proto