Use pytest's vendored TerminalWriter and pathlib. Replace old use of pyimport. Ref #226.

Author: ionelmc

setup.py

@@ -69,7 +69,6 @@ def read(*names, **kwargs):
     install_requires=[
         'pytest>=3.8',
         'py-cpuinfo',
-        'py>=1.8.2',
     ],
     extras_require={
         'aspect': ['aspectlib'],

src/pytest_benchmark/cli.py

@@ -1,7 +1,8 @@
 import argparse
 from functools import partial
 
-import py
+from _pytest import pathlib
+from _pytest._io import TerminalWriter
 
 from pytest_benchmark.csv import CSVResults
 
@@ -118,13 +119,15 @@ def make_parser():
 
 
 class HookDispatch(object):
-    def __init__(self):
-        conftest_file = py.path.local('conftest.py')
-        if conftest_file.check():
-            self.conftest = conftest_file.pyimport()
+    def __init__(self, **kwargs):
+        conftest_file = pathlib.Path('conftest.py')
+        if conftest_file.exists():
+            self.conftest = pathlib.import_path(conftest_file, **kwargs)
         else:
             self.conftest = None
 
+        print(f'>>>>>>>>>> {self.conftest}')
+
     def __getattr__(self, item):
         default = getattr(plugin, item)
         return getattr(self.conftest, item, default)
@@ -139,7 +142,7 @@ def main():
     logger = Logger(level)
     storage = load_storage(args.storage, logger=logger, netrc=args.netrc)
 
-    hook = HookDispatch()
+    hook = HookDispatch(mode=args.importmode, root=pathlib.Path('.'))
 
     if args.command == 'list':
         for file in storage.query():
@@ -172,7 +175,7 @@ def main():
 
 class TerminalReporter(object):
     def __init__(self):
-        self._tw = py.io.TerminalWriter()
+        self._tw = TerminalWriter()
 
     def ensure_newline(self):
         pass
@@ -181,8 +184,8 @@ def write(self, content, **markup):
         self._tw.write(content, **markup)
 
     def write_line(self, line, **markup):
-        if not py.builtin._istext(line):
-            line = py.builtin.text(line, errors="replace")
+        if not isinstance(line, str):
+            line = line.decode(errors="replace")
         self._tw.line(line, **markup)
 
     def rewrite(self, line, **markup):

src/pytest_benchmark/logger.py

@@ -4,7 +4,7 @@
 import sys
 import warnings
 
-import py
+from _pytest._io import TerminalWriter
 from pytest import PytestWarning
 
 
@@ -17,7 +17,7 @@ class Logger(object):
 
     def __init__(self, level=NORMAL, config=None):
         self.level = level
-        self.term = py.io.TerminalWriter(file=sys.stderr)
+        self.term = TerminalWriter(file=sys.stderr)
         self.suspend_capture = None
         self.resume_capture = None
         if config:

src/pytest_benchmark/plugin.py

@@ -126,6 +126,14 @@ def add_global_options(addoption, prefix="benchmark-"):
         action="store_true", default=False,
         help="Disable reporting. Verbose mode takes precedence."
     )
+    if not prefix:
+        addoption(
+            "--import-mode",
+            default="prepend",
+            choices=["prepend", "append", "importlib"],
+            dest="importmode",
+            help="How to attempt loading hooks from conftest. Akin to pytest's --import-mode. Default: %(default)r.",
+        )
 
 
 def pytest_addoption(parser):