From 868e5b831b27ac07cb4a73e04ed9e6fac5987fc7 Mon Sep 17 00:00:00 2001
From: Brad Warren <bmw@users.noreply.github.com>
Date: Tue, 5 Jun 2018 17:59:11 -0700
Subject: [PATCH] Make python setup.py test use pytest for acme (#6072)

---
 acme/setup.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/acme/setup.py b/acme/setup.py
index f8196e953..0fb4d8fff 100644
--- a/acme/setup.py
+++ b/acme/setup.py
@@ -1,6 +1,7 @@
 from setuptools import setup
 from setuptools import find_packages
-
+from setuptools.command.test import test as TestCommand
+import sys
 
 version = '0.25.0.dev0'
 
@@ -33,6 +34,19 @@ docs_extras = [
     'sphinx_rtd_theme',
 ]
 
+class PyTest(TestCommand):
+    user_options = []
+
+    def initialize_options(self):
+        TestCommand.initialize_options(self)
+        self.pytest_args = ''
+
+    def run_tests(self):
+        import shlex
+        # import here, cause outside the eggs aren't loaded
+        import pytest
+        errno = pytest.main(shlex.split(self.pytest_args))
+        sys.exit(errno)
 
 setup(
     name='acme',
@@ -65,5 +79,7 @@ setup(
         'dev': dev_extras,
         'docs': docs_extras,
     },
+    tests_require=["pytest"],
     test_suite='acme',
+    cmdclass={"test": PyTest},
 )