Allow for specifying local wheels and sdists as dependencies (#215)

It is useful, especially when creating custom versions of dependencies that are compatible with CPython main, to specify local file paths as dependencies. Unfortunately, that's blocked by a feature that treats any dependency that matches a file path as a requirements file. This makes that check a little more specific which allows for installing wheels.

Author: mdboom

pyperformance/_pip.py

@@ -150,7 +150,7 @@ def install_requirements(reqs, *extra,
     if upgrade:
         args.append('-U')  # --upgrade
     for reqs in [reqs, *extra]:
-        if os.path.exists(reqs):
+        if os.path.isfile(reqs) and reqs.endswith('.txt'):
             args.append('-r')  # --requirement
         args.append(reqs)
     return run_pip('install', *args, **kwargs)

pyperformance/tests/data/MANIFEST

@@ -0,0 +1,4 @@
+[benchmarks]
+
+name	metafile
+local_wheel	<local>

pyperformance/tests/data/bm_local_wheel/pyproject.toml

@@ -0,0 +1,9 @@
+[project]
+name = "pyperformance_bm_local_wheel"
+requires-python = ">=3.8"
+dependencies = ["pyperf"]
+urls = {repository = "https://github.com/python/pyperformance"}
+version = "1.0"
+
+[tool.pyperformance]
+name = "local_wheel"

pyperformance/tests/data/bm_local_wheel/requirements.txt

@@ -0,0 +1 @@
+this_is-1.0.2-py2.py3-none-any.whl#egg=this_is

pyperformance/tests/data/bm_local_wheel/run_benchmark.py

@@ -0,0 +1,19 @@
+"""
+A dummy benchmark that uses a local wheel
+"""
+
+import pyperf
+from this_is import en_us
+
+
+def bench():
+    return 1.0
+
+
+if __name__ == "__main__":
+    runner = pyperf.Runner()
+    runner.metadata['description'] = "A dummy benchmark that has a local wheel dependency"
+
+    args = runner.parse_args()
+    runner.bench_func('local_wheel', bench)
+

pyperformance/tests/data/bm_local_wheel/this_is-1.0.2-py2.py3-none-any.whl

@@ -161,6 +161,19 @@ def test_run_and_show(self):
         # XXX Capture and check the output.
         self.run_module('pyperf', 'slowest', filename)
 
+    def test_run_test_benchmarks(self):
+        # Run the benchmarks that exist only for testing
+        # in pyperformance/tests/data
+        filename = self.resolve_tmp('bench-test.json')
+
+        self.run_pyperformance(
+            'run',
+            '--manifest', os.path.join(tests.DATA_DIR, 'MANIFEST'),
+            '-b', 'all',
+            '-o', filename,
+            capture=None,
+        )
+
     ###################################
     # compile
 

pyperformance/tests/test_commands.py

@@ -44,7 +44,11 @@ def _add_from_file(self, filename):
         if not os.path.exists(filename):
             return
         for line in _utils.iter_clean_lines(filename):
-            self._add(line)
+            fullpath = os.path.join(os.path.dirname(filename), line.strip())
+            if os.path.isfile(fullpath):
+                self._add(fullpath)
+            else:
+                self._add(line)
 
     def _add(self, line):
         self.specs.append(line)