Test pickle python version.

MAIF · Mar 15, 2024 · 81c5397 · 81c5397
1 parent dd04cd1
commit 81c5397
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/unit_tests/explainer/test_smart_explainer.py b/tests/unit_tests/explainer/test_smart_explainer.py
@@ -1,6 +1,7 @@
 """
 Unit test for smart explainer
 """
+
 import os
 import sys
 import types
@@ -633,14 +634,14 @@ def test_load_1(self):
         temp, xpl = init_sme_to_pickle_test()
 
         current = Path(path.abspath(__file__)).parent.parent.parent
-        if str(sys.version)[0:3] == "3.8":
-            pkl_file = path.join(current, "data/xpl_to_load_38.pkl")
-        elif str(sys.version)[0:3] == "3.9":
+        if str(sys.version)[0:3] == "3.9":
             pkl_file = path.join(current, "data/xpl_to_load_39.pkl")
         elif str(sys.version)[0:4] == "3.10":
             pkl_file = path.join(current, "data/xpl_to_load_310.pkl")
         elif str(sys.version)[0:4] == "3.11":
             pkl_file = path.join(current, "data/xpl_to_load_311.pkl")
+        elif str(sys.version)[0:4] == "3.12":
+            pkl_file = path.join(current, "data/xpl_to_load_312.pkl")
         else:
             raise NotImplementedError
 

diff --git a/tests/unit_tests/utils/test_load_smartpredictor.py b/tests/unit_tests/utils/test_load_smartpredictor.py
@@ -1,6 +1,7 @@
 """
 Unit test smart predictor
 """
+
 import sys
 import unittest
 from os import path
@@ -29,12 +30,12 @@ def test_load_smartpredictor_1(self):
         current = Path(path.abspath(__file__)).parent.parent.parent
         if str(sys.version)[0:4] == "3.10":
             pkl_file = path.join(current, "data/predictor_to_load_310.pkl")
-        elif str(sys.version)[0:3] == "3.8":
-            pkl_file = path.join(current, "data/predictor_to_load_38.pkl")
         elif str(sys.version)[0:3] == "3.9":
             pkl_file = path.join(current, "data/predictor_to_load_39.pkl")
         elif str(sys.version)[0:4] == "3.11":
             pkl_file = path.join(current, "data/predictor_to_load_311.pkl")
+        elif str(sys.version)[0:4] == "3.12":
+            pkl_file = path.join(current, "data/predictor_to_load_312.pkl")
         else:
             raise NotImplementedError