Make paths OS-agnostic

Author: chriscarrollsmith

codegraph/utils.py

@@ -3,7 +3,7 @@
 from typing import List, Union
 
 
-def get_python_paths_list(paths: Union[str, List]) -> List:
+def get_python_paths_list(paths: Union[str, List]) -> List[str]:
     """
         return list of paths to python files, that found in provided path
     :param paths: paths to folder or python file that need to tests
@@ -22,11 +22,10 @@ def get_python_paths_list(paths: Union[str, List]) -> List:
     for path in paths:
         path = Path(path).absolute()
         if not path.exists():
-            raise ValueError(f"Path {path.as_posix()} does not exists")
-        path = path.as_posix()
+            raise ValueError(f"Path {path.as_posix()} does not exist")
         paths_list += [
-            path
-            for path in glob.glob(path + "/*", recursive=True)
-            if path.endswith(".py") and not path.endswith("__init__.py")
+            Path(p).as_posix()
+            for p in glob.glob(str(path / "**" / "*.py"), recursive=True)
+            if not p.endswith("__init__.py")
         ]
     return paths_list

tests/test_utils.py

@@ -12,13 +12,10 @@ def test_get_python_paths_list_error():
 
 
 def test_get_python_paths_list():
-    expected = list(
-        map(
-            lambda x: (pathlib.Path(__file__).parents[1] / "codegraph" / x).as_posix(),
-            ["core.py", "parser.py", "utils.py", "vizualyzer.py", "main.py"],
-        )
-    )
-    assert sorted(
-        get_python_paths_list(
-            (pathlib.Path(__file__).parents[1] / "codegraph").as_posix()
-        )) == sorted(expected)
+    base_path = pathlib.Path(__file__).parents[1] / "codegraph"
+    expected = [
+        (base_path / x).as_posix()
+        for x in ["core.py", "parser.py", "utils.py", "vizualyzer.py", "main.py"]
+    ]
+    result = get_python_paths_list(base_path.as_posix())
+    assert sorted(result) == sorted(expected)