trying to fix setup

Author: rhayes777

release.sh

@@ -21,7 +21,9 @@ set +e
 git commit -m "Incremented version number"
 set -e
 
-python setup.py test sdist bdist_wheel
+pytest $p
+
+python setup.py sdist bdist_wheel
 twine upload dist/* --skip-existing --username $PYPI_USERNAME --password $PYPI_PASSWORD
 
 # docker login -u $DOCKER_USERNAME -p $DOCKER_PASSWORD

setup.py

@@ -1,10 +1,30 @@
 from codecs import open
 from os.path import abspath, dirname, join
+from subprocess import call
 
-from setuptools import find_packages, setup
+from setuptools import find_packages, setup, Command
 
 from autoarray import __version__
 
+
+class RunTests(Command):
+    """Run all tests."""
+
+    description = "run tests"
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        """Run all tests!"""
+        errno = call(["py.test", "--cov=autolens", "--cov-report=term-missing"])
+        raise SystemExit(errno)
+
+
 this_dir = abspath(dirname(__file__))
 with open(join(this_dir, "README.md"), encoding="utf-8") as file:
     long_description = file.read()
@@ -40,6 +60,6 @@
     keywords="cli",
     packages=find_packages(exclude=["docs"]),
     install_requires=requirements,
-    setup_requires=["pytest-runner"],
-    tests_require=["pytest"],
+    extras_require={"test": ["coverage", "pytest", "pytest-cov"]},
+    cmd_class={"test": RunTests}
 )

test_autoarray/conftest.py

@@ -1,8 +1,7 @@
-import os
-
 import pytest
 from matplotlib import pyplot
 
+
 class PlotPatch(object):
     def __init__(self):
         self.paths = []