Fix cmdtest --test

1 files changed, 7 insertions, 0 deletions

diff --git a/cmdtest b/cmdtest
index b8c6efa..d7f329e 100755
--- a/cmdtest
+++ b/cmdtest
@@ -120,6 +120,13 @@ class CommandTester(cliapp.Application):
         for test in self.settings['test']:
             matches = glob.glob(os.path.join(dirname, test + '.*'))
             filenames += [os.path.basename(x) for x in matches]
+
+        basenames = ['setup_once', 'setup', 'teardown', 'teardown_once']
+        for basename in basenames:
+            filename = os.path.join(dirname, basename)
+            if os.path.exists(filename):
+                filenames.append(basename)
+
         return filenames
 
     def setup_tempdir(self):