Simplify setup.py

Author: rth

examples/simple-extension/setup.py

@@ -1,34 +1,9 @@
 import sys
 from setuptools import find_packages, setup
-from setuptools.command.test import test as TestCommand
 from setuptools_rust import RustExtension
 
 
-class PyTest(TestCommand):
-    user_options = []
-
-    def run(self):
-        self.run_command("test_rust")
-        import subprocess
-        errno = subprocess.call(['pytest', 'tests'])
-        raise SystemExit(errno)
-
-
-def get_cfg_flags():
-    version = sys.version_info[0:2]
-    if version[0] == 2:
-        return ['--cfg=Py_2']
-    else:
-        return ['--cfg=Py_3']
-
-
-def get_features():
-    version = sys.version_info[0:2]
-    if version[0] == 2:
-        return ['numpy/python2']
-    else:
-        return ['numpy/python3']
-
+PYTHON_MAJOR_VERSION = sys.version_info[0]
 
 setup_requires = ['setuptools-rust>=0.6.0']
 install_requires = ['numpy']
@@ -41,13 +16,12 @@ def get_features():
     rust_extensions=[RustExtension(
         'rust_ext.rust_ext',
         './Cargo.toml',
-        rustc_flags=get_cfg_flags(),
-        features=get_features(),
+        rustc_flags=['--cfg=Py_{}'.format(PYTHON_MAJOR_VERSION)],
+        features=['numpy/python{}'.format(PYTHON_MAJOR_VERSION)],
     )],
     install_requires=install_requires,
     setup_requires=setup_requires,
     test_requires=test_requires,
     packages=find_packages(),
     zip_safe=False,
-    cmdclass=dict(test=PyTest)
 )

examples/simple-extension/tox.ini

@@ -10,5 +10,4 @@ skip_missing_interpreters = true
 description = Run the unit tests under {basepython}
 deps = -rrequirements-dev.txt
 usedevelop = True
-commands = pip install -e .
-           pytest
+commands = pytest