Wheels 3 (#1764)

1 files changed, 8 insertions, 3 deletions

diff --git a/setup.py b/setup.py
index ac7b8b53..3c442302 100755
--- a/setup.py
+++ b/setup.py
@@ -47,6 +47,7 @@ from _compat import PY3  # NOQA
 from _compat import which  # NOQA
 
 
+PYPY = '__pypy__' in sys.builtin_module_names
 macros = []
 if POSIX:
     macros.append(("PSUTIL_POSIX", 1))
@@ -68,11 +69,15 @@ if POSIX:
 
 
 extras_require = {"test": [
+    "enum34; python_version <= '3.4'",
     "ipaddress; python_version < '3.0'",
     "mock; python_version < '3.0'",
-    "pypiwin32; sys.platform == 'win32'",
-    "wmi; sys.platform == 'win32'",
-    "enum34; python_version <= '3.4'"]}
+    "unittest2; python_version < '3.0'",
+]}
+if not PYPY:
+    extras_require['test'].extend([
+        "pywin32; sys.platform == 'win32'",
+        "wmi; sys.platform == 'win32'"])
 
 
 def get_version():