umami.tests.unit.metrics package#

Submodules#

umami.tests.unit.metrics.test_metrics module#

Unit test script for the functions in metrics.py.

class umami.tests.unit.metrics.test_metrics.CalcDiscValuesTestCase(methodName='runTest')#

Bases: TestCase

Test class for the calc_disc_values function.

setUp()#: Hook method for setting up the test fixture before exercising it.

test_calc_disc_values()#: Test calc_disc_values for three classes and bjets (main).

test_calc_disc_values_2signal_classes_bkg_jets()#: Test calc_disc_values for four classes and 2 signal classes and calculation of scores for background class (not-main class).

test_calc_disc_values_2singal_classes()#: Test calc_disc_values for three classes and cjets (not-main class).

test_calc_disc_values_error()#: Testing raising of errors.

test_calc_disc_values_rejection()#: Test calc_disc_values for three classes and cjets (not-main class).

class umami.tests.unit.metrics.test_metrics.GetRejectionTestCase(methodName='runTest')#

Bases: TestCase

Test class for the GetRejection function.

setUp()#: Create a default dataset for testing.

test_get_rejection()#: Test GetRejection for three classes.

test_get_rejection_4_classes()#: Test GetRejection for four classes.

test_get_rejection_wrong_shapes()#: Test GetRejection with incorrect shapes in y_pred and y_true. Also checking if incorrect shapes between y_* and class_labels.

class umami.tests.unit.metrics.test_metrics.GetScoreTestCase(methodName='runTest')#

Bases: TestCase

Test class for the GetScore function.

setUp()#: Create a default dataset for testing.

test_get_score()#: Test GetScore for three classes.

test_get_score_4classes()#: Test GetScore for four classes.

test_get_score_errors()#: Test error raising of get_score

class umami.tests.unit.metrics.test_metrics.MetricsSmallTestCase(methodName='runTest')#

Bases: TestCase

Test class for the umami.metrics functions.

setUp()#: Hook method for setting up the test fixture before exercising it.

test_discriminant_output_shape()#: Test the discriminant output shape layer.

Module contents#