def testFindHistogramLevelSparseDiagnostics(self):
histogram = histogram_module.Histogram('hist', 'count')
histograms = histogram_module.HistogramSet([histogram])
histograms.AddSharedDiagnostic('foo', histogram_module.Generic('bar'))
histograms.AddSharedDiagnostic(
'telemetry', histogram_module.TelemetryInfo())
diagnostics = add_histograms.FindHistogramLevelSparseDiagnostics(
histogram.guid, histograms)
self.assertEqual(1, len(diagnostics))
self.assertIsInstance(diagnostics[0], histogram_module.TelemetryInfo)
def testFindHistogramLevelSparseDiagnostics(self):
hist = histogram_module.Histogram('hist', 'count')
histograms = histogram_set.HistogramSet([hist])
histograms.AddSharedDiagnostic('foo',
histogram_module.GenericSet(['bar']))
histograms.AddSharedDiagnostic(reserved_infos.GPUS.name,
histogram_module.GenericSet([]))
diagnostics = add_histograms.FindHistogramLevelSparseDiagnostics(
hist.guid, histograms)
self.assertEqual(1, len(diagnostics))
self.assertIsInstance(diagnostics[0], histogram_module.GenericSet)