umami.tests.unit.data_tools package#

Submodules#

umami.tests.unit.data_tools.test_data_tools_cuts module#

Test scripts for the data tools cuts.

class umami.tests.unit.data_tools.test_data_tools_cuts.GetCategoryCutsTestCase(methodName='runTest')#

Bases: TestCase

Test the implementation of the get_category_cuts function.

setUp() → None#

Hook method for setting up the test fixture before exercising it.

test_float_case()#

Testing floats.

test_integer_case()#

Testing ints.

test_list_case()#

Testing lists.

test_wrong_type_provided()#

Testing wrong type provided.

class umami.tests.unit.data_tools.test_data_tools_cuts.GetSampleCutsTestCase(methodName='runTest')#

Bases: TestCase

Test the implementation of the get_sample_cuts function.

setUp()#

Hook method for setting up the test fixture before exercising it.

test_cuts_passing_ttbar()#

Testing cuts for ttbar pass.

test_cuts_passing_ttbar_extended_labelling()#

Testing cuts for ttbar pass with the extended labelling.

test_cuts_wrong_format()#

Testing raise of KeyError for wrongly formatted cuts.

test_cuts_wrong_modulo()#

Testing raise of KeyError for wrong modulo cuts.

test_cuts_wrong_operator()#

Testing raise of KeyError for wrong operator cuts.

umami.tests.unit.data_tools.test_data_tools_loaders module#

Test scripts for the data tools loaders.

class umami.tests.unit.data_tools.test_data_tools_loaders.LoadFilesTestCase(methodName='runTest')#

Bases: TestCase

Test class for the different loading functions.

setUp()#

Hook method for setting up the test fixture before exercising it.

test_cut_operators()#

Test different operators for the cuts.

test_cut_vars()#

Test cut vars.

test_different_chunk_size()#

Test different chunk size.

test_h5_errors()#

Test the raise of errors for wrong h5 file.

test_load_jets_from_file()#

Test the loading of the jet variables from file.

test_load_trks_from_file()#

Test the correct loading of the track variables.

test_trks_different_chunk_size()#

Test different chunk size.

test_trks_different_operators()#

Test different operators.

test_trks_raise_errors()#

Test the raise of the different errors.

test_wrong_filepath()#

Test raise of error for wrong filepath.

umami.tests.unit.data_tools.test_data_tools_tools module#

Test scripts for the data tools tools.

class umami.tests.unit.data_tools.test_data_tools_tools.CompareH5FilesVariablesTestCase(methodName='runTest')#

Bases: TestCase

Test the implementation of the compare_h5_files_variables function.

setUp() → None#

Hook method for setting up the test fixture before exercising it.

test_common_vars()#

Test the common vars are equal.

test_common_vars_3_files()#

Test the common vars for 3 files.

test_different_vars()#

Test the case where variables differ between 2 files.

test_different_vars_3_files()#

Test the case where variables differ between 3 files.

test_no_inputs()#

Test the ValueError raise when no input is given.

Module contents#