diff --git a/tests/unit_test.py b/tests/unit_test.py index bd4e9cb..6c91a2c 100644 --- 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'))