From 7118458d0300a85ef3ff278923b2656dd2d04be8 Mon Sep 17 00:00:00 2001 From: gonfeco Date: Thu, 26 Oct 2023 14:09:58 +0200 Subject: [PATCH] Fixed metric error in tests --- tests/test_btc_02_ae.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_btc_02_ae.py b/tests/test_btc_02_ae.py index 7fee052..f2839df 100644 --- a/tests/test_btc_02_ae.py +++ b/tests/test_btc_02_ae.py @@ -84,7 +84,7 @@ def test_ae_iqae(): a = pd.read_csv(filename, header=[0, 1], index_col=[0, 1]) #print(a["absolute_error_sum"]["mean"]) #print(100* list(a['KS']['mean'])[0]) - assert(100* list(a['absolute_error_sum']['mean'])[0] < 1.0) + assert(100* list(a['IntegralAbsoluteError']['mean'])[0] < 1.0) shutil.rmtree(folder) def test_ae_mlae(): @@ -134,7 +134,7 @@ def test_ae_mlae(): a = pd.read_csv(filename, header=[0, 1], index_col=[0, 1]) #print(a["absolute_error_sum"]["mean"]) #print(100* list(a['KS']['mean'])[0]) - assert(100* list(a['absolute_error_sum']['mean'])[0] < 1.0) + assert(100* list(a['IntegralAbsoluteError']['mean'])[0] < 1.0) shutil.rmtree(folder) def test_ae_rqae(): @@ -181,7 +181,7 @@ def test_ae_rqae(): filename = folder + benchmark_arguments["summary_results"] a = pd.read_csv(filename, header=[0, 1], index_col=[0, 1]) print(a["absolute_error_sum"]["mean"]) - assert((a["absolute_error_sum"]["mean"] < 0.01).all()) + assert((a["IntegralAbsoluteError"]["mean"] < 0.01).all()) #print(100* list(a['KS']['mean'])[0]) #assert(100* list(a['absolute_error_sum']['mean'])[0] < 1.0) shutil.rmtree(folder)