summaryrefslogtreecommitdiff
path: root/setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'setup.py')
-rw-r--r--setup.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index eb3d9102..ad2891c1 100644
--- a/setup.py
+++ b/setup.py
@@ -12,6 +12,7 @@
# pylint: disable=W0404,W0622,W0613
"""Setup script for astroid."""
import os
+import sys
from setuptools import find_packages, setup
from setuptools.command import easy_install # pylint: disable=unused-import
from setuptools.command import install_lib # pylint: disable=unused-import
@@ -27,6 +28,9 @@ with open(os.path.join(astroid_dir, "README.rst")) as fobj:
long_description = fobj.read()
+needs_pytest = set(['pytest', 'test', 'ptr']).intersection(sys.argv)
+pytest_runner = ['pytest-runner'] if needs_pytest else []
+
def install():
return setup(
name="astroid",
@@ -42,7 +46,7 @@ def install():
install_requires=install_requires,
extras_require=extras_require,
packages=find_packages() + ["astroid.brain"],
- setup_requires=["pytest-runner"],
+ setup_requires=pytest_runner,
test_suite="test",
tests_require=["pytest"],
)