umami.tests.unit.plotting_tools package#

Submodules#

umami.tests.unit.plotting_tools.test_eval_plot_functions module#

Unit test script for eval plot functions.

class umami.tests.unit.plotting_tools.test_eval_plot_functions.EvalPlotsTestCase(methodName='runTest')#

Bases: TestCase

Test class for the eval plot functions.

setUp()#

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

test_plot_fraction_contour()#

Test the plot_fraction_contour function.

test_plot_prob()#

Test nominal behaviour for the probability plots without ratio.

test_plot_prob_comparison()#

Test nominal behaviour for the probability plots with ratio.

test_plot_pt_dependence()#

Test nominal for pT vs rejection.

test_plot_roc()#

Test nominal for roc with one rejection.

test_plot_roc_comparison()#

Test nominal for roc with two rejection.

test_plot_saliency()#

Test nominal behaviour for the saliency plots.

test_plot_score()#

Test nominal behaviour for plot_score without ratio.

test_plot_score_comparison()#

Test nominal behaviour for plot_score with ratio.

umami.tests.unit.plotting_tools.test_plotting_utils module#

Unit test script for utils functions of the plotting functions.

class umami.tests.unit.plotting_tools.test_plotting_utils.RetrieveTruthLabelVarValueTestCase(methodName='runTest')#

Bases: TestCase

Test class for the retrieve_truth_label_var_value function.

setUp() → None#

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

test_3_classes()#

Test nomimal behaviour for 3 classes without extended labeling

test_4_classes()#

Test nomimal behaviour for 4 classes without extended labeling

umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions module#

Unit test script for the preprocessing plot functions.

class umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PlotResamplingVariablesTestCase(methodName='runTest')#

Bases: TestCase

Test class for the plot resampling variables functions.

setUp()#

Get dataset for testing.

test_plot_resampling_variables()#

Test the plot_resampling_variables

test_plot_resampling_variables_wrong_binning()#

Test the plot_resampling_variables wrong binning error

class umami.tests.unit.plotting_tools.test_preprocessing_plotting_functions.PreprocessingPlotsTestCase(methodName='runTest')#

Bases: TestCase

Test class for the preprocessing plot functions.

setUp()#

Get dataset for testing.

test_plot_unique_jet_appearence()#

Testing the plot unique jet appearance nominal behaviour.

test_preprocessing_plots()#

Test nominal behaviour for the preprocessing plots.

umami.tests.unit.plotting_tools.test_train_plot_functions module#

Unit test script for the train plot functions.

class umami.tests.unit.plotting_tools.test_train_plot_functions.GetCompTaggerRejDictTestCase(methodName='runTest')#

Bases: TestCase

Test the rejection dict calculation.

setUp()#

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

test_get_comp_tagger_rej_dict_dips_dl1()#

Test get_comp_tagger_rej_dict for 1 taggers.

test_get_comp_tagger_rej_dict_errors()#

Test get_comp_tagger_rej_dict error raising.

test_get_comp_tagger_rej_dict_umami()#

Test get_comp_tagger_rej_dict for 2 taggers.

class umami.tests.unit.plotting_tools.test_train_plot_functions.TrainPlotsTestCase(methodName='runTest')#

Bases: TestCase

Test class for the train plot function.

setUp()#

Create a default dataset for testing.

test_plot_accuracies()#

Test the accuracy plot.

test_plot_accuracies_umami()#

Test the accuracy plot with two taggers.

test_plot_disc_cut_per_epoch()#

Test the discriminant cut per epoch plot.

test_plot_disc_cut_per_epoch_umami()#

Test the discriminant cut per epoch plot with two taggers.

test_plot_losses()#

Test the loss plot.

test_plot_losses_umami()#

Test the loss plot with two taggers.

test_plot_rej_per_epoch()#

Test the rejection per epoch plot.

test_plot_rej_per_epoch_comp()#

Test the rejection per epoch comparison plot.

Module contents#