7 from quantity_extract
import RootQuantityExtract, default_extractor
13 Various test cases for the Quantity Extractor class
18 Test getting the quantities of a TH1d
21 h1 = ROOT.TH1D(
"ext_test1",
"ext_test1", 40, 0, 40.0)
27 res = rext.extract(h1)
29 self.assertIn(
"mean_x", res)
30 self.assertAlmostEqual(res[
"mean_x"], 20)
31 self.assertIn(
"mean_y", res)
32 self.assertAlmostEqual(res[
"mean_y"], 0.075)
33 self.assertIn(
"mean_y_zero_suppressed", res)
34 self.assertAlmostEqual(res[
"mean_y_zero_suppressed"], 1.0)
35 self.assertIn(
"entries", res)
39 Test getting the quantities of a TProfile
42 h1 = ROOT.TProfile(
"ext_test1",
"ext_test1", 40, 0, 40.0)
48 res = rext.extract(h1)
50 self.assertIn(
"mean_y", res)
51 self.assertAlmostEqual(res[
"mean_y"], 0.375)
52 self.assertIn(
"mean_y_zero_suppressed", res)
53 self.assertAlmostEqual(res[
"mean_y_zero_suppressed"], 5.0)
57 Test getting the quanities contained in a TNtuple
60 tn = ROOT.TNtuple(
"particle_list",
"particle_list",
"x:y:z:energy")
64 res = rext.extract(tn)
65 self.assertEqual(len(res), 4)
66 self.assertIn(
"particle_list_z", res)
67 self.assertAlmostEqual(res[
"particle_list_z"], 7.0)
70 if __name__ ==
"__main__":