diff --git a/Test/compare_test.py b/Test/compare_test.py index 71b55912..7902de90 100644 --- a/Test/compare_test.py +++ b/Test/compare_test.py @@ -122,13 +122,13 @@ >>> cm_null = ConfusionMatrix(matrix={0: {0: 0, 1: 0, 2: 0}, 1: {0: 0, 1: 0, 2: 0}, 2: {0: 0, 1: 0, 2: 0}}) >>> with warns(RuntimeWarning, match='Confusion matrices are too similar to identify a clear best option.'): ... cp8 = Compare({"cm1": cm, "cm2": cm}, class_weight={'L3': 6, 'L1': 3, 'L2': 3}) ->>> with warns(RuntimeWarning, match='Invalid class_weight format; the result is for unweighted mode.'): +>>> with warns(RuntimeWarning, match='Invalid `class_weight` format; the result is for unweighted mode.'): ... cp9 = Compare({"cm1": cm1, "cm2": cm2}, class_weight={0: 0, 1: 0, 2: 0}) >>> class_benchmark_weight = {"PLRI": 0, "NLRI": 0, "DPI": 0, "AUCI": 0, "MCCI": 0, "QI": 0} ->>> with warns(RuntimeWarning, match='Invalid class_benchmark_weight format; the result is for unweighted mode.'): +>>> with warns(RuntimeWarning, match='Invalid `class_benchmark_weight` format; the result is for unweighted mode.'): ... cp10 = Compare({"cm1": cm1, "cm2": cm2}, class_benchmark_weight=class_benchmark_weight) >>> overall_benchmark_weight = {"SOA1": 0, "SOA2": 0, "SOA3": 0, "SOA4": 0, "SOA5": 0, "SOA6": 0, "SOA7": 0, "SOA8": 0, "SOA9": 0, "SOA10": 0} ->>> with warns(RuntimeWarning, match='Invalid overall_benchmark_weight format; the result is for unweighted mode.'): +>>> with warns(RuntimeWarning, match='Invalid `overall_benchmark_weight` format; the result is for unweighted mode.'): ... cp11 = Compare({"cm1": cm1, "cm2": cm2}, overall_benchmark_weight=overall_benchmark_weight) >>> with warns(RuntimeWarning, match='Confusion matrices are too similar to identify a clear best option.'): ... cp12 = Compare({"cm1": cm_null, "cm2": cm_null})