From 419c899c944203a3556f31fe9c0d79b0ced7f2bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Mon, 16 Dec 2024 00:53:11 +0100
Subject: [PATCH] Fix some errors in tests

---
 benchmark/Datasets.py                                   | 2 +-
 benchmark/tests/Models_test.py                          | 2 +-
 benchmark/tests/datasets/all.txt                        | 4 ++--
 benchmark/tests/test_files/exreport_excel_Datasets.test | 2 +-
 benchmark/tests/test_files/report_datasets.test         | 2 +-
 5 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/benchmark/Datasets.py b/benchmark/Datasets.py
index 4909f0c..b2458c4 100644
--- a/benchmark/Datasets.py
+++ b/benchmark/Datasets.py
@@ -152,7 +152,7 @@ class Datasets:
         if dataset_name is None:
             return class_names, results
         try:
-            class_name = class_names[sets.index(dataset_name)]
+            class_name = class_names[results.index(dataset_name)]
         except ValueError:
             raise ValueError(f"Unknown dataset: {dataset_name}")
         return [class_name], [dataset_name]
diff --git a/benchmark/tests/Models_test.py b/benchmark/tests/Models_test.py
index e804f95..446b0ef 100644
--- a/benchmark/tests/Models_test.py
+++ b/benchmark/tests/Models_test.py
@@ -102,7 +102,7 @@ class ModelTest(TestBase):
         test = {
             "STree": ((11, 6, 4), 1.0),
             "Wodt": ((303, 152, 50), 0.9382022471910112),
-            "ODTE": ((7.86, 4.43, 3.37), 1.0),
+            "ODTE": ((786, 443, 337), 1.0),
             "Cart": ((23, 12, 5), 1.0),
             "SVC": ((0, 0, 0), 0.7078651685393258),
             "RandomForest": ((21.3, 11, 5.26), 1.0),
diff --git a/benchmark/tests/datasets/all.txt b/benchmark/tests/datasets/all.txt
index 48584fd..9da1d4a 100644
--- a/benchmark/tests/datasets/all.txt
+++ b/benchmark/tests/datasets/all.txt
@@ -1,2 +1,2 @@
-iris,class,all
-wine,class,[0, 1]
+iris;class;all
+wine;class;[0, 1]
diff --git a/benchmark/tests/test_files/exreport_excel_Datasets.test b/benchmark/tests/test_files/exreport_excel_Datasets.test
index b3cdca0..1d07ff9 100644
--- a/benchmark/tests/test_files/exreport_excel_Datasets.test
+++ b/benchmark/tests/test_files/exreport_excel_Datasets.test
@@ -1,4 +1,4 @@
-1;1;"Datasets used in benchmark ver. 0.5.0"
+1;1;"Datasets used in benchmark ver. 1.0.1"
 2;1;" Default score accuracy"
 2;2;"Cross validation"
 2;6;"5 Folds"
diff --git a/benchmark/tests/test_files/report_datasets.test b/benchmark/tests/test_files/report_datasets.test
index 5e6d4c8..ac35256 100644
--- a/benchmark/tests/test_files/report_datasets.test
+++ b/benchmark/tests/test_files/report_datasets.test
@@ -1,4 +1,4 @@
-Datasets used in benchmark ver. 0.5.0
+Datasets used in benchmark ver. 1.0.1
 
 Dataset                        Sampl. Feat. Cont Cls Balance
 ============================== ====== ===== ==== === ==========================================