From bb452457d6a3d4cc1119cddcd64172f0458bd653 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Sat, 26 Apr 2014 16:42:33 +0100
Subject: Refactor setup.py: simpler method for running tests

---
 setup.py | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

(limited to 'setup.py')

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',
-- 
cgit v1.2.1