Update conditional setup-time dependency to also support pytest and test commands.

1 files changed, 2 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 807c0180..e1b302a3 100755
--- a/setup.py
+++ b/setup.py
@@ -75,7 +75,8 @@ if sys.platform == 'win32' or force_windows_specific_files:
     package_data.setdefault('setuptools', []).extend(['*.exe'])
     package_data.setdefault('setuptools.command', []).extend(['*.xml'])
 
-pytest_runner = ['pytest-runner'] if 'ptr' in sys.argv else []
+needs_pytest = set(['ptr', 'pytest', 'test']) in sys.argv else []
+pytest_runner = ['pytest-runner'] if needs_pytest else []
 needs_sphinx = set(['build_sphinx', 'upload_docs']).intersection(sys.argv)
 sphinx = ['sphinx', 'rst.linker'] if needs_sphinx else []