Revert "fix: remove test command and related code from setup.py"

This reverts commit 9a8366f.

Author: pyrooka

setup.py

@@ -17,6 +17,7 @@
 import sys
 import pkg_resources
 from setuptools import setup, find_packages
+from setuptools.command.test import test as TestCommand
 
 __version__ = '3.20.3'
 
@@ -35,6 +36,22 @@
 with open('requirements-dev.txt', encoding='utf-8') as f:
     tests_require = [str(req) for req in pkg_resources.parse_requirements(f)]
 
+
+class PyTest(TestCommand):
+    """PyTest class."""
+
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = ['--strict', '--verbose', '--tb=long', 'test']
+        self.test_suite = True
+
+    def run_tests(self):
+        import pytest
+
+        errcode = pytest.main(self.test_args)
+        sys.exit(errcode)
+
+
 with open("README.md", "r", encoding='utf-8') as fh:
     readme = fh.read()
 
@@ -45,6 +62,7 @@
     license='Apache 2.0',
     install_requires=install_requires,
     tests_require=tests_require,
+    cmdclass={'test': PyTest},
     author='IBM',
     author_email='[email protected]',
     long_description=readme,