umami.tests.unit package#
Subpackages#
- umami.tests.unit.data_tools package
- Submodules
- umami.tests.unit.data_tools.test_data_tools_cuts module
- umami.tests.unit.data_tools.test_data_tools_loaders module
LoadFilesTestCase
LoadFilesTestCase.setUp()
LoadFilesTestCase.test_cut_operators()
LoadFilesTestCase.test_cut_vars()
LoadFilesTestCase.test_different_chunk_size()
LoadFilesTestCase.test_h5_errors()
LoadFilesTestCase.test_load_jets_from_file()
LoadFilesTestCase.test_load_trks_from_file()
LoadFilesTestCase.test_trks_different_chunk_size()
LoadFilesTestCase.test_trks_different_operators()
LoadFilesTestCase.test_trks_raise_errors()
LoadFilesTestCase.test_wrong_filepath()
- umami.tests.unit.data_tools.test_data_tools_tools module
CompareH5FilesVariablesTestCase
CompareH5FilesVariablesTestCase.setUp()
CompareH5FilesVariablesTestCase.test_common_vars()
CompareH5FilesVariablesTestCase.test_common_vars_3_files()
CompareH5FilesVariablesTestCase.test_different_vars()
CompareH5FilesVariablesTestCase.test_different_vars_3_files()
CompareH5FilesVariablesTestCase.test_no_inputs()
- Module contents
- umami.tests.unit.evaluation_tools package
- umami.tests.unit.helper_tools package
- Submodules
- umami.tests.unit.helper_tools.test_classification_tools module
GetClassTestCase
GetClassTestCase.setUp()
GetClassTestCase.test_get_class_label_variables_3_classes()
GetClassTestCase.test_get_class_label_variables_4_classes()
GetClassTestCase.test_get_class_prob_var_names_3_classes()
GetClassTestCase.test_get_class_prob_var_names_4_classes()
GetClassTestCase.test_get_cut_list_3_classes()
GetClassTestCase.test_get_cut_list_4_classes()
- umami.tests.unit.helper_tools.test_histogram_tools module
- umami.tests.unit.helper_tools.test_tools module
- Module contents
- umami.tests.unit.input_vars_tools package
- Submodules
- umami.tests.unit.input_vars_tools.test_input_vars_tools module
HelperFunctionTestCase
JetPlottingTestCase
JetPlottingTestCase.setUp()
JetPlottingTestCase.test_plot_input_vars_jets()
JetPlottingTestCase.test_plot_input_vars_jets_comparison()
JetPlottingTestCase.test_plot_input_vars_jets_comparison_wrong_type()
JetPlottingTestCase.test_plot_input_vars_jets_log_value_error()
JetPlottingTestCase.test_plot_input_vars_jets_wrong_type()
JetPlottingTestCase.test_plot_input_vars_trks()
JetPlottingTestCase.test_plot_input_vars_trks_comparison()
JetPlottingTestCase.test_plot_input_vars_trks_comparison_not_normalised()
JetPlottingTestCase.test_plot_input_vars_trks_comparison_wrong_type()
JetPlottingTestCase.test_plot_input_vars_trks_log_error_raise()
JetPlottingTestCase.test_plot_input_vars_trks_wrong_type()
JetPlottingTestCase.test_plot_n_tracks_per_jet()
- Module contents
- umami.tests.unit.metrics package
- Submodules
- umami.tests.unit.metrics.test_metrics module
CalcDiscValuesTestCase
CalcDiscValuesTestCase.setUp()
CalcDiscValuesTestCase.test_calc_disc_values()
CalcDiscValuesTestCase.test_calc_disc_values_2signal_classes_bkg_jets()
CalcDiscValuesTestCase.test_calc_disc_values_2singal_classes()
CalcDiscValuesTestCase.test_calc_disc_values_error()
CalcDiscValuesTestCase.test_calc_disc_values_rejection()
GetRejectionTestCase
GetScoreTestCase
MetricsSmallTestCase
- Module contents
- umami.tests.unit.plotting_tools package
- Submodules
- umami.tests.unit.plotting_tools.test_eval_plot_functions module
EvalPlotsTestCase
EvalPlotsTestCase.setUp()
EvalPlotsTestCase.test_plot_fraction_contour()
EvalPlotsTestCase.test_plot_prob()
EvalPlotsTestCase.test_plot_prob_comparison()
EvalPlotsTestCase.test_plot_pt_dependence()
EvalPlotsTestCase.test_plot_roc()
EvalPlotsTestCase.test_plot_roc_comparison()
EvalPlotsTestCase.test_plot_saliency()
EvalPlotsTestCase.test_plot_score()
EvalPlotsTestCase.test_plot_score_comparison()
- umami.tests.unit.plotting_tools.test_plotting_utils module
- umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions module
- umami.tests.unit.plotting_tools.test_train_plot_functions module
GetCompTaggerRejDictTestCase
TrainPlotsTestCase
TrainPlotsTestCase.setUp()
TrainPlotsTestCase.test_plot_accuracies()
TrainPlotsTestCase.test_plot_accuracies_umami()
TrainPlotsTestCase.test_plot_disc_cut_per_epoch()
TrainPlotsTestCase.test_plot_disc_cut_per_epoch_umami()
TrainPlotsTestCase.test_plot_losses()
TrainPlotsTestCase.test_plot_losses_umami()
TrainPlotsTestCase.test_plot_rej_per_epoch()
TrainPlotsTestCase.test_plot_rej_per_epoch_comp()
- Module contents
- umami.tests.unit.preprocessing package
- Submodules
- umami.tests.unit.preprocessing.test_preprocessing module
- umami.tests.unit.preprocessing.test_preprocessing_tools module
BinariseJetLabelsTestCase
BinariseJetLabelsTestCase.setUp()
BinariseJetLabelsTestCase.test_2_classes_correct_output()
BinariseJetLabelsTestCase.test_correct_output()
BinariseJetLabelsTestCase.test_missing_internal_label()
BinariseJetLabelsTestCase.test_shape_2_classes()
BinariseJetLabelsTestCase.test_shape_array()
BinariseJetLabelsTestCase.test_shape_data_frame()
BinariseJetLabelsTestCase.test_wrong_type()
BinariseJetLabelsTestCase.test_zero_length()
CreateSamplesTestCase
GetDictTestCase
GetVariableDictTestCase
PrepareSamplesTestCase
PrepareSamplesTestCase.CArgs
PrepareSamplesTestCase.setUp()
PrepareSamplesTestCase.test_get_batches_per_file()
PrepareSamplesTestCase.test_jets_generator_fullcuts()
PrepareSamplesTestCase.test_jets_generator_fullcuts_wotracks()
PrepareSamplesTestCase.test_jets_generator_lightcut()
PrepareSamplesTestCase.test_no_sample_provided()
PrepareSamplesTestCase.test_run()
PrepareSamplesTestCase.test_wrong_category_provided()
PrepareSamplesTestCase.test_wrong_sample_provided()
TTbarMergeTestCase
- umami.tests.unit.preprocessing.test_preprocessing_tools_configuration module
PreparationTestCase
PreprocessConfigurationTestCase
PreprocessConfigurationTestCase.setUp()
PreprocessConfigurationTestCase.test_genera_settings_class_plot_options_as_dict()
PreprocessConfigurationTestCase.test_general_settings_class_as_dict()
PreprocessConfigurationTestCase.test_get_file_name_no_input()
PreprocessConfigurationTestCase.test_get_file_name_no_iterations()
PreprocessConfigurationTestCase.test_get_file_name_no_iterations_no_input()
PreprocessConfigurationTestCase.test_missing_key_error()
PreprocessConfigurationTestCase.test_sampling_class_as_dict()
PreprocessConfigurationTestCase.test_sampling_options_class_as_dict()
- umami.tests.unit.preprocessing.test_preprocessing_tools_merge_large_h5 module
- umami.tests.unit.preprocessing.test_preprocessing_tools_resampling module
CalculateBinningTestCase
CorrectFractionsTestCase
CorrectFractionsTestCase.test_different_input_length_class_names()
CorrectFractionsTestCase.test_different_input_lengths()
CorrectFractionsTestCase.test_input_correct_fractions()
CorrectFractionsTestCase.test_not_fraction_sum_one()
CorrectFractionsTestCase.test_scaling_down_largest()
CorrectFractionsTestCase.test_scaling_down_small()
CorrectFractionsTestCase.test_twice_same_fractions()
CorrectFractionsTestCase.test_zero_length()
CorrectFractionsTestCase.test_zero_n_jets()
PDFResamplingTestCase
ResamplingTestCase
ResamplingTestCaseComplexDistr
ResamplingTestCaseEasyDistr
SamplingGeneratorTestCase
UnderSamplingNoReplaceTestCase
UnderSamplingTestCase
UnderSamplingTestCaseEasyDistr
- umami.tests.unit.preprocessing.test_preprocessing_tools_scaling module
- Module contents
- umami.tests.unit.tf_tools package
- Submodules
- umami.tests.unit.tf_tools.test_generators module
BaseGeneratorTest
TestCadsGenerator
TestDL1Generator
TestDipsGenerator
TestModelGenerator
TestModelGenerator.test_init()
TestModelGenerator.test_init_no_n_jets_no_x_given()
TestModelGenerator.test_init_no_n_jets_x_name_given()
TestModelGenerator.test_init_no_n_jets_x_trk_name_given()
TestModelGenerator.test_load_in_memory()
TestModelGenerator.test_load_in_memory_x_name_error()
TestModelGenerator.test_load_in_memory_x_trk_name_error()
TestUmamiCondAttGenerator
TestUmamiGenerator
- umami.tests.unit.tf_tools.test_tddgenerators module
TestTDDGenerator
TestTDDGenerator.setUp()
TestTDDGenerator.tearDown()
TestTDDGenerator.test_circular_modelgen_eq_tddgen()
TestTDDGenerator.test_circular_modelgen_eq_tddgen_old_scaling()
TestTDDGenerator.test_circular_with_excluded_var()
TestTDDGenerator.test_circular_with_weights_in_tdd()
TestTDDGenerator.test_get_generator()
- umami.tests.unit.tf_tools.test_tf_tools module
AttentionTestCase
AttentionTestCase.setUp()
AttentionTestCase.test_call()
AttentionTestCase.test_call_assertion_error()
AttentionTestCase.test_call_no_softmax()
AttentionTestCase.test_call_with_mask()
AttentionTestCase.test_compute_mask()
AttentionTestCase.test_compute_mask_errors()
AttentionTestCase.test_get_config()
ConditionalDeepSetTestCase
ConfigObject
DeepSetTestCase
DenseNetTestCase
PrepareModelTestCase
PrepareModelTestCase.setUp()
PrepareModelTestCase.test_init_fresh_model()
PrepareModelTestCase.test_init_fresh_model_no_load_optimiser_given()
PrepareModelTestCase.test_load_model_with_continue_training()
PrepareModelTestCase.test_load_model_without_continue_training()
PrepareModelTestCase.test_load_optimiser_value_rrror()
- umami.tests.unit.tf_tools.test_tf_tools_convert_to_record module
- Module contents
- umami.tests.unit.tools package
- umami.tests.unit.train_tools package
- Submodules
- umami.tests.unit.train_tools.test_nn_tools module
CallbackBaseTestCase
ConfigObject
ConfigurationTestCase
ConfigurationTestCase.setUp()
ConfigurationTestCase.test_cads_config()
ConfigurationTestCase.test_cads_without_cond_info()
ConfigurationTestCase.test_dips_att_with_cond_info()
ConfigurationTestCase.test_dips_config()
ConfigurationTestCase.test_double_defined_b_jets()
ConfigurationTestCase.test_double_defined_c_jets()
ConfigurationTestCase.test_double_label_value()
ConfigurationTestCase.test_missing_key_error()
ConfigurationTestCase.test_no_batch_size()
ConfigurationTestCase.test_no_eval_batch_size()
ConfigurationTestCase.test_no_val_batch_size()
ConfigurationTestCase.test_no_val_no_eval_batch_size()
ConfigurationTestCase.test_umami_config()
CreateMetadataFolderTestCase
GetDropoutRatesTestCase
GetEpochFromStringTestCase
GetJetFeatureIndicesTestCase
GetJetFeaturePositionTestCase
GetMetricsFileNameTestCase
GetModelPathTestCase
GetSamplesTestCase
GetSamplesTestCase.setUp()
GetSamplesTestCase.test_get_test_file()
GetSamplesTestCase.test_get_test_sample()
GetSamplesTestCase.test_get_test_sample_errors()
GetSamplesTestCase.test_get_test_sample_extended_labeling()
GetSamplesTestCase.test_get_test_sample_trks()
GetSamplesTestCase.test_get_test_sample_trks_extended_labeling()
GetSamplesTestCase.test_load_validation_data()
GetSamplesTestCase.test_load_validation_data_no_var_cuts()
GetSamplesTestCase.test_load_validation_data_unsupported_tagger()
GetUniqueIdentifiersTestCase
MinimalPreprocessingConfig
MyCallbackTestCase
MyCallbackUmamiTestCase
SetupOutputDirectoryTestCase
- Module contents