fix unit test

aimclub · Jun 26, 2024 · 38ee957 · 38ee957
1 parent 6c7df50
commit 38ee957
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit_test.py b/tests/unit_test.py
@@ -1,6 +1,7 @@
 import unittest
 import os
 import sys
+from pathlib import Path
 project_dir = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.dirname(os.path.dirname(project_dir)))
 from ODRS.src.data_processing.ml_processing.recommendation_module import predict_models
@@ -24,7 +25,7 @@ def test_data_path(self):
         default = '/home/runner/work/ODRS/ODRS/user_datasets/WaRP/Warp-D'
         ROOT = '/home/runner/work/ODRS/ODRS'
         result = get_data_path(ROOT, default)
-        self.assertEqual(result, '/home/runner/work/ODRS/ODRS/user_datasets/Warp-D')
+        self.assertEqual(result, Path('/home/runner/work/ODRS/ODRS/user_datasets/Warp-D'))