Refactor setup.py: simpler method for running tests

1 files changed, 14 insertions, 9 deletions

diff --git a/setup.py b/setup.py
index b3c7ff1..807b46a 100644
--- a/setup.py
+++ b/setup.py
@@ -50,17 +50,22 @@ class Check(Command):
 
     def run(self):
         if self.unit_tests:
-            print "run unit tests"
-            cliapp.runcmd(
-                ['python', '-m', 'CoverageTestRunner',
-                 '--ignore-missing-from=without-tests'],
-                stdout=None, stderr=None)
-            os.remove('.coverage')
+            self.run_unit_tests()
 
         if self.yarns:
-            cliapp.runcmd(
-                ['yarn'] + glob.glob('yarns/*.yarn'),
-                stdout=None, stderr=None)
+            self.run_yarns()
+
+    def run_unit_tests(self):
+        cliapp.runcmd(
+            ['python', '-m', 'CoverageTestRunner',
+             '--ignore-missing-from=without-tests'],
+            stdout=None, stderr=None)
+        os.remove('.coverage')
+
+    def print_yarns(self):
+        cliapp.runcmd(
+            ['yarn'] + glob.glob('yarns/*.yarn'),
+            stdout=None, stderr=None)
 
 
 setup(name='distix',