mva_examples.py

#!/usr/bin/env python3
# -- analysis/tests/mva_examples.py
 
 
 
import b2test_utils as b2tu
import b2test_utils_analysis as b2tua
 
 
 
 
class ExamplesTest(b2tua.ExamplesTest):
    def test_mva_examples(self):
        """
        Test supported mva examples.
        """
        # list of examples that cannot be tested
        broken_mva_egs = ['B2A712-DeepContinuumSuppression_MVATrain.py',  # training takes too long and cannot be shortened
                          'B2A713-DeepContinuumSuppression_MVAExpert.py',  # requires weight file created in previous example
                          'B2A714-DeepContinuumSuppression_MVAModel.py'  # MVA model description and not standalone example script
                          ]
 
        self._test_examples_dir('analysis/examples/mva/', broken_mva_egs, nevents=100)
 
 
if __name__ == '__main__':
    with b2tu.clean_working_directory():
        b2tua.unittest.main()