def test_data_for_th1f(self):
"""Appropriate information is provided on TH1F ROOT objects."""
data = tasks.data_for_object(self.dummy_th1)
for key in ('binning', 'values', 'uncertainties', 'axis_titles'):
assert key in data
nbins = self.dummy_th1.GetNbinsX()
assert len(data['binning']) == nbins
assert len(data['values']) == nbins
assert len(data['uncertainties']) == nbins
assert len(data['axis_titles']) == 2
assert data['axis_titles'][0] == self.dummy_th1.GetXaxis().GetTitle()
assert data['axis_titles'][1] == self.dummy_th1.GetYaxis().GetTitle()
def test_data_for_unknown_tobject(self):
"""No information should be returned for unsupported ROOT objects."""
# Use a child of TObject that we don't support
obj = ROOT.TH3D()
data = tasks.data_for_object(obj)
assert not data
