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)
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