use pytest in setuptools (sonic-net#490)

lguohan · Mar 22, 2019 · 1b734f1 · 1b734f1
1 parent 7030c97
commit 1b734f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -0,0 +1,2 @@
+[aliases]
+test=pytest
diff --git a/setup.py b/setup.py
@@ -16,12 +16,6 @@
 
 import glob
 from setuptools import setup
-import unittest
-
-def get_test_suite():
-    test_loader = unittest.TestLoader()
-    test_suite = test_loader.discover('sonic-utilities-tests', pattern='*.py')
-    return test_suite
 
 setup(
     name='sonic-utilities',
@@ -125,7 +119,11 @@ def get_test_suite():
         'click',
         'natsort'
     ],
+    setup_requires= [
+        'pytest-runner'
+    ],
     tests_require = [
+        'pytest',
         'mock>=2.0.0',
         'mockredispy>=2.9.3'
     ],