From 5bec9d4d2f55bfb138b5274b2e5370a29fe5c10d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Fri, 8 Mar 2024 13:22:31 +0100
Subject: [PATCH] Update test

---
 tests/TestPythonClassifiers.cc | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tests/TestPythonClassifiers.cc b/tests/TestPythonClassifiers.cc
index beee7d6..efd08cf 100644
--- a/tests/TestPythonClassifiers.cc
+++ b/tests/TestPythonClassifiers.cc
@@ -34,6 +34,12 @@ TEST_CASE("Test Python Classifiers score", "[PyClassifiers]")
         {"SVC", new pywrap::SVC()},
         {"RandomForest", new pywrap::RandomForest()}
     };
+    map<std::string, std::string> versions = {
+        {"ODTE", "0.3.6"},
+        {"STree", "1.3.2"},
+        {"SVC", "1.3.2"},
+        {"RandomForest", "1.3.2"}
+    };
     auto clf = models[name];
 
     SECTION("Test Python Classifier " + name + " score ")
@@ -49,7 +55,7 @@ TEST_CASE("Test Python Classifiers score", "[PyClassifiers]")
     SECTION("Library check version")
     {
         INFO("Checking version of " + name + " classifier");
-        REQUIRE(clf->getVersion() == ACTUAL_VERSION);
+        REQUIRE(clf->getVersion() == versions[name]);
     }
 }
 TEST_CASE("Classifiers features", "[PyClassifiers]")