TestStdCharged Class Reference

Inheritance diagram for TestStdCharged:

Public Member Functions
def	test_nonesense_list (self)
def	test_all_list (self)
def	test_good_list (self)
def	test_higheff (self)
def	test_loose (self)
def	test_percentile_eff (self)
def	test_mostLikely_lists (self)

Protected Member Functions
def	_check_list_name (self, target, functionname, particleloader)
def	_check_listtype_exists (self, listtype, functions=_base_functions)
def	_check_function_call (self, functions=_base_functions, expectedlist=stdCharged._defaultlist)
def	_check_listtype_does_not_exist (self, listtype, functions=_base_functions)

Detailed Description

Test case for charged standard particle lists

Definition at line 25 of file test_std_charged.py.

Member Function Documentation

_check_function_call()

def _check_function_call (   self,   functions = _base_functions,   expectedlist = stdCharged._defaultlist  )

protected

check that a function works (i.e. adds a particle loader)

Definition at line 56 of file test_std_charged.py.

    def _check_function_call(self, functions=_base_functions, expectedlist=stdCharged._defaultlist):
        """check that a function works (i.e. adds a particle loader)"""
        for f in functions:
            testpath = create_path()
            f(path=testpath)
            self.assertEqual(
                len(testpath.modules()), 3,
                f"Function {f.__name__} doesn't work")
            self.assertTrue(any(module.type() == "ParticleLoader" for module in testpath.modules()))
            loader = testpath.modules()[0]
            self._check_list_name(expectedlist, f.__name__, loader)
 

_check_list_name()

def _check_list_name (   self,   target,   functionname,   particleloader  )

protected

Check that the list name we expect is the one that was
actually added to the ParticleLoader

Definition at line 28 of file test_std_charged.py.

    def _check_list_name(self, target, functionname, particleloader):
        """
        Check that the list name we expect is the one that was
        actually added to the ParticleLoader
        """
        for param in particleloader.available_params():
            if param.name == 'decayStrings':
                name = param.values[0].split(':')[1]
                self.assertTrue(
                    name == target,
                    "Name: \"%s\" added by function %s, expecting \"%s\""
                    % (name, functionname, target))
 

_check_listtype_does_not_exist()

def _check_listtype_does_not_exist (   self,   listtype,   functions = _base_functions  )

protected

check that a given listtype function doesn't do anything

Definition at line 68 of file test_std_charged.py.

    def _check_listtype_does_not_exist(self, listtype, functions=_base_functions):
        """check that a given listtype function doesn't do anything"""
        for f in functions:
            testpath = create_path()
            f(listtype, path=testpath)
            self.assertEqual(
                len(testpath.modules()), 0,
                f"List {listtype} works with function {f.__name__}")
            self.assertFalse(any(module.type() == "ParticleLoader" for module in testpath.modules()))
 

_check_listtype_exists()

def _check_listtype_exists (   self,   listtype,   functions = _base_functions  )

protected

check that a given listtype function works

Definition at line 41 of file test_std_charged.py.

    def _check_listtype_exists(self, listtype, functions=_base_functions):
        """check that a given listtype function works"""
        for f in functions:
            testpath = create_path()
            f(listtype, path=testpath)
            self.assertEqual(
                len(testpath.modules()), 1 if listtype == 'all' else 3,
                f"List {listtype} doesn't work with function {f.__name__}")
            self.assertTrue(any(module.type() == "ParticleLoader" for module in testpath.modules()))
 
            # now we're certain that the loader is the only module on the path,
            # so check it was added with the ParticleList name we expect
            loader = testpath.modules()[0]
            self._check_list_name(listtype, f.__name__, loader)
 

test_all_list()

def test_all_list

(

self

)

check that the builder functions all work with the all list

Definition at line 82 of file test_std_charged.py.

    def test_all_list(self):
        """check that the builder functions all work with the all list"""
        self._check_listtype_exists("all")
 

test_good_list()

def test_good_list

(

self

)

check that the builder functions all work with the good list

Definition at line 86 of file test_std_charged.py.

    def test_good_list(self):
        """check that the builder functions all work with the good list"""
        self._check_listtype_exists("good")
        # all functions may be called with no arguments: makes a "good" list
        self._check_function_call(_base_functions)
 

test_higheff()

def test_higheff

(

self

)

check that the builder functions all work with the higheff list

Definition at line 92 of file test_std_charged.py.

    def test_higheff(self):
        """check that the builder functions all work with the higheff list"""
        self._check_listtype_exists("higheff")
 

test_loose()

def test_loose

(

self

)

check that the builder functions all work with the loose list

Definition at line 96 of file test_std_charged.py.

    def test_loose(self):
        """check that the builder functions all work with the loose list"""
        self._check_listtype_exists("loose")
 

test_mostLikely_lists()

def test_mostLikely_lists

(

self

)

check that the builder functions work with the mostLikely lists

Definition at line 117 of file test_std_charged.py.

    def test_mostLikely_lists(self):
        """check that the builder functions work with the mostLikely lists"""
        nLists = 5  # Number of ParticleLoader's expected
        testpath = create_path()
        stdCharged.stdMostLikely(path=testpath)
        self.assertEqual(
            len(testpath.modules()), 3 * nLists,
            f"There should be {int(nLists)} fillParticleList calls")
        self.assertTrue(any(module.type() == "ParticleLoader" for module in testpath.modules()))
        for module in testpath.modules():
            self._check_list_name('mostlikely', 'stdMostLikely', module)
 
 

test_nonesense_list()

def test_nonesense_list

(

self

)

check that the builder functions fail with a nonexisting list

Definition at line 78 of file test_std_charged.py.

    def test_nonesense_list(self):
        """check that the builder functions fail with a nonexisting list"""
        self._check_listtype_does_not_exist("flibble")
 

test_percentile_eff()

def test_percentile_eff

(

self

)

check that the builder functions all work with the percentile eff lists

Definition at line 100 of file test_std_charged.py.

    def test_percentile_eff(self):
        """check that the builder functions all work with the percentile eff lists"""
        for function in _base_functions:
            eff_exists = 0
            for ename in stdCharged._effnames:
                cut = stdCharged._stdChargedEffCuts(stdCharged._chargednames[_base_functions.index(function)],
                                                    ename)
                if 0.0 < cut < 1.0:
                    self._check_listtype_exists(ename, [function])
                    eff_exists += 1
                else:
                    self._check_listtype_does_not_exist(ename, [function])
            self.assertTrue(
                eff_exists,
                "Function: \"%s\" has no valid list based on efficiency percentile."
                % (function.__name__))
 

---

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

• analysis/tests/test_std_charged.py