def test_training_time_saved(self): self.assertEqual(metrics.GetTrainingTimeSaved(api_label="baz"), 0) metrics.AddTrainingTimeSaved(api_label="baz", microseconds=1000) self.assertEqual(metrics.GetTrainingTimeSaved(api_label="baz"), 1000)
def _get_time_saved(self, api_label): return metrics.GetTrainingTimeSaved(api_label=api_label)