pkg_resources to pathlib etc.

Author: soodoku

allocator/tests/test_cluster_kahip.py

@@ -8,13 +8,13 @@
 
 import os
 import unittest
-from pkg_resources import resource_filename
+from pathlib import Path
 
 from allocator.cluster_kahip import main
 from . import capture
 
-
-ROADS = resource_filename(__name__, "chonburi-roads-50.csv")
+TEST_DIR = Path(__file__).parent
+ROADS = str(TEST_DIR / "chonburi-roads-50.csv")
 
 
 def kahip_available():

allocator/tests/test_sort_by_distance.py

@@ -7,13 +7,14 @@
 """
 
 import unittest
-from pkg_resources import resource_filename
+from pathlib import Path
 
 from allocator.sort_by_distance import main
 from . import capture
 
-CENTROIDS = resource_filename(__name__, "worker-locations.csv")
-ROADS = resource_filename(__name__, "chonburi-roads-50.csv")
+TEST_DIR = Path(__file__).parent
+CENTROIDS = str(TEST_DIR / "worker-locations.csv")
+ROADS = str(TEST_DIR / "chonburi-roads-50.csv")
 
 
 class TestSortByDistance(unittest.TestCase):

pyproject.toml

@@ -107,28 +107,6 @@ python_files = ["test_*.py"]
 python_classes = ["Test*"]
 python_functions = ["test_*"]
 
-[tool.flake8]
-max-line-length = 127
-max-complexity = 10
-select = ["E9", "F63", "F7", "F82"]
-ignore = ["E203", "W503"]
-exclude = [
-    ".git",
-    "__pycache__",
-    ".venv",
-    "venv",
-    ".env",
-    "env",
-    "dist",
-    "build",
-    "*.egg-info",
-    ".tox",
-    ".coverage",
-    ".coverage.*",
-    ".cache",
-    ".mypy_cache",
-    ".pytest_cache"
-]
 
 [tool.coverage.run]
 source = ["allocator"]