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
LoadFilesTestCaseLoadFilesTestCase.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
CompareH5FilesVariablesTestCaseCompareH5FilesVariablesTestCase.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
GetClassTestCaseGetClassTestCase.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
HelperFunctionTestCaseJetPlottingTestCaseJetPlottingTestCase.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
CalcDiscValuesTestCaseCalcDiscValuesTestCase.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()
GetRejectionTestCaseGetScoreTestCaseMetricsSmallTestCase
- Module contents
- umami.tests.unit.plotting_tools package
- Submodules
- umami.tests.unit.plotting_tools.test_eval_plot_functions module
EvalPlotsTestCaseEvalPlotsTestCase.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
GetCompTaggerRejDictTestCaseTrainPlotsTestCaseTrainPlotsTestCase.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
BinariseJetLabelsTestCaseBinariseJetLabelsTestCase.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()
CreateSamplesTestCaseGetDictTestCaseGetVariableDictTestCasePrepareSamplesTestCasePrepareSamplesTestCase.CArgsPrepareSamplesTestCase.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
PreparationTestCasePreprocessConfigurationTestCasePreprocessConfigurationTestCase.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
CalculateBinningTestCaseCorrectFractionsTestCaseCorrectFractionsTestCase.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()
PDFResamplingTestCaseResamplingTestCaseResamplingTestCaseComplexDistrResamplingTestCaseEasyDistrSamplingGeneratorTestCaseUnderSamplingNoReplaceTestCaseUnderSamplingTestCaseUnderSamplingTestCaseEasyDistr
- 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
BaseGeneratorTestTestCadsGeneratorTestDL1GeneratorTestDipsGeneratorTestModelGeneratorTestModelGenerator.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()
TestUmamiCondAttGeneratorTestUmamiGenerator
- umami.tests.unit.tf_tools.test_tddgenerators module
TestTDDGeneratorTestTDDGenerator.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
AttentionTestCaseAttentionTestCase.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()
ConditionalDeepSetTestCaseConfigObjectDeepSetTestCaseDenseNetTestCasePrepareModelTestCasePrepareModelTestCase.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
CallbackBaseTestCaseConfigObjectConfigurationTestCaseConfigurationTestCase.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()
CreateMetadataFolderTestCaseGetDropoutRatesTestCaseGetEpochFromStringTestCaseGetJetFeatureIndicesTestCaseGetJetFeaturePositionTestCaseGetMetricsFileNameTestCaseGetModelPathTestCaseGetSamplesTestCaseGetSamplesTestCase.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()
GetUniqueIdentifiersTestCaseMinimalPreprocessingConfigMyCallbackTestCaseMyCallbackUmamiTestCaseSetupOutputDirectoryTestCase
- Module contents