# You should have received a copy of the GNU General Public License
 # along with LinPy.  If not, see <http://www.gnu.org/licenses/>.
 
-from distutils.core import setup, Extension
+from setuptools import setup, Extension
 
 
 with open('linpy/_version.py') as file:
     author='MINES ParisTech',
     url='https://scm.cri.mines-paristech.fr/git/linpy.git',
     packages=['linpy'],
-    ext_modules = [
+    ext_modules=[
         Extension('linpy._islhelper',
             sources=['linpy/_islhelper.c'],
             libraries=['isl'])
-    ]
+    ],
+    test_suite='linpy.tests'
 )