Fix issues

PickwickSoft · Jul 31, 2023 · 4446271 · 4446271
1 parent bd92446
commit 4446271
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pystreamapi/loaders/__csv_loader.py b/pystreamapi/loaders/__csv_loader.py
@@ -17,9 +17,7 @@ def csv(file_path: str, delimiter=',', encoding="utf-8") -> LazyFileIterable:
         :param delimiter: The delimiter used in the CSV file.
     """
     file_path = __validate_path(file_path)
-    loader = lambda: __load_csv(file_path, delimiter, encoding)
-
-    return LazyFileIterable(loader)
+    return LazyFileIterable(lambda: __load_csv(file_path, delimiter, encoding))
 
 
 def __load_csv(file_path, delimiter, encoding):

diff --git a/tests/test_loaders.py b/tests/test_loaders.py
@@ -38,6 +38,6 @@ def test_csv_loader_with_invalid_path(self):
         with self.assertRaises(FileNotFoundError):
             csv(f'{self.path}/invalid.csv')
 
-    def test_csv_loader_with_non_absolute_path(self):
+    def test_csv_loader_with_non_file(self):
         with self.assertRaises(ValueError):
-            csv('invalid.csv')
+            csv(f'{self.path}/')