fix: test case on windows

Sinaptik-AI · Jan 8, 2025 · 3982c45 · 3982c45
1 parent 7e207eb
commit 3982c45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit_tests/dataframe/test_loader.py b/tests/unit_tests/dataframe/test_loader.py
@@ -120,7 +120,7 @@ def test_get_cache_file_path_with_destination_path(self, sample_schema):
         loader.dataset_path = "test/users"
         cache_path = loader._get_cache_file_path()
         if sys.platform.startswith("win"):
-            assert cache_path.endswith("datasets\\test\\users\\users.parquet")
+            assert cache_path.endswith("datasets\\test/users\\users.parquet")
         else:
             assert cache_path.endswith("datasets/test/users/users.parquet")
 
@@ -132,7 +132,7 @@ def test_get_cache_file_path_without_destination_path(self, sample_schema):
         loader.dataset_path = "test/users"
         cache_path = loader._get_cache_file_path()
         if sys.platform.startswith("win"):
-            assert cache_path.endswith("datasets\\test\\users\\data.parquet")
+            assert cache_path.endswith("datasets\\test/users\\data.parquet")
         else:
             assert cache_path.endswith("datasets/test/users/data.parquet")