TestQuantityExtract Class Reference

Inheritance diagram for TestQuantityExtract:

Public Member Functions
def	test_h1d (self)
def	test_profile (self)
def	test_ntuple (self)

Detailed Description

Various test cases for the Quantity Extractor class

Definition at line 17 of file test_quantity_extract.py.

Member Function Documentation

test_h1d()

def test_h1d

(

self

)

Test getting the quantities of a TH1d

Definition at line 23 of file test_quantity_extract.py.

    def test_h1d(self):
        """
        Test getting the quantities of a TH1d
        """
 
        h1 = ROOT.TH1D("ext_test1", "ext_test1", 40, 0, 40.0)
        h1.Fill(10.0)
        h1.Fill(20.0)
        h1.Fill(30.0)
 
        rext = RootQuantityExtract(default_extractor())
        res = rext.extract(h1)
 
        self.assertIn("mean_x", res)
        self.assertAlmostEqual(res["mean_x"], 20)
        self.assertIn("mean_y", res)
        self.assertAlmostEqual(res["mean_y"], 0.075)
        self.assertIn("mean_y_zero_suppressed", res)
        self.assertAlmostEqual(res["mean_y_zero_suppressed"], 1.0)
        self.assertIn("entries", res)
 

test_ntuple()

def test_ntuple

(

self

)

Test getting the quanities contained in a TNtuple

Definition at line 62 of file test_quantity_extract.py.

    def test_ntuple(self):
        """
        Test getting the quanities contained in a TNtuple
        """
 
        tn = ROOT.TNtuple("particle_list", "particle_list", "x:y:z:energy")
        tn.Fill(5, 6, 7, 10)
 
        rext = RootQuantityExtract(default_extractor())
        res = rext.extract(tn)
        self.assertEqual(len(res), 4)
        self.assertIn("particle_list_z", res)
        self.assertAlmostEqual(res["particle_list_z"], 7.0)
 
 

test_profile()

def test_profile

(

self

)

Test getting the quantities of a TProfile

Definition at line 44 of file test_quantity_extract.py.

    def test_profile(self):
        """
        Test getting the quantities of a TProfile
        """
 
        h1 = ROOT.TProfile("ext_test1", "ext_test1", 40, 0, 40.0)
        h1.Fill(10.0, 4.0)
        h1.Fill(20.0, 5.0)
        h1.Fill(30.0, 6.0)
 
        rext = RootQuantityExtract(default_extractor())
        res = rext.extract(h1)
 
        self.assertIn("mean_y", res)
        self.assertAlmostEqual(res["mean_y"], 0.375)
        self.assertIn("mean_y_zero_suppressed", res)
        self.assertAlmostEqual(res["mean_y_zero_suppressed"], 5.0)
 

---

The documentation for this class was generated from the following file:

• validation/tests/test_quantity_extract.py