allow running python setup.py test

ilanschnell · Feb 20, 2023 · 9fdee1a · 9fdee1a
1 parent c991d40
commit 9fdee1a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/Makefile b/Makefile
@@ -6,7 +6,7 @@ bitarray/_bitarray.so: bitarray/_bitarray.c
 
 
 test: bitarray/_bitarray.so
-	$(PYTHON) -c "import bitarray; bitarray.test()"
+	$(PYTHON) setup.py test
 
 
 install:

diff --git a/setup.py b/setup.py
@@ -1,13 +1,16 @@
 import re
 import sys
+
+if "test" in sys.argv:
+    import bitarray
+    # when test was successful, return 0 (hence not)
+    sys.exit(not bitarray.test().wasSuccessful())
+
 try:
     from setuptools import setup, Extension
 except ImportError:
     from distutils.core import setup, Extension
 
-if "test" in sys.argv:
-    exit('ERROR: To run the bitarray test suite, type:\n\n'
-         '    python -c "import bitarray; bitarray.test()"\n')
 
 kwds = {}
 try: