Enable 'setup.py test'

Author: 1st1

setup.py

@@ -10,6 +10,7 @@
 import platform
 import re
 import sys
+import unittest
 
 import setuptools
 from setuptools.command import build_ext as _build_ext
@@ -26,6 +27,12 @@
     LDFLAGS.append('ws2_32.lib')
 
 
+def discover_tests():
+    test_loader = unittest.TestLoader()
+    test_suite = test_loader.discover('tests', pattern='test_*.py')
+    return test_suite
+
+
 class build_ext(_build_ext.build_ext):
     user_options = _build_ext.build_ext.user_options + [
         ('cython-always', None,
@@ -193,4 +200,5 @@ def _patch_cfile(self, cfile):
             extra_link_args=LDFLAGS)
     ],
     cmdclass={'build_ext': build_ext},
+    test_suite='setup.discover_tests',
 )