1 files changed, 10 insertions, 10 deletions
diff --git a/tests/test_all.py b/tests/test_all.py
index 8ae1c76..fec4ae4 100644
--- a/tests/test_all.py
+++ b/tests/test_all.py
@@ -1,14 +1,14 @@
 
-import unittest
-
-def main():
-    import yaml
-    names = ['test_yaml']
-    if yaml.__libyaml__:
-        names.append('test_yaml_ext')
-    suite = unittest.defaultTestLoader.loadTestsFromNames(names)
-    runner = unittest.TextTestRunner()
-    runner.run(suite)
+import sys, yaml, test_appliance
+
+def main(args=None):
+    collections = []
+    import test_yaml
+    collections.append(test_yaml)
+    if yaml.__with_libyaml__:
+        import test_yaml_ext
+        collections.append(test_yaml_ext)
+    test_appliance.run(collections, args)
 
 if __name__ == '__main__':
     main()