diff --git a/test/test_plugins.py b/test/test_plugins.py index a7590432f..a48dd10ae 100644 --- a/test/test_plugins.py +++ b/test/test_plugins.py @@ -66,3 +66,10 @@ class PluginTest(unittest.TestCase, TestHelper): def register_plugin(self, plugin_class): self._plugin_classes.add(plugin_class) + + +def suite(): + return unittest.TestLoader().loadTestsFromName(__name__) + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/test/test_types_plugin.py b/test/test_types_plugin.py index c7f1f4383..d4d9a96ef 100644 --- a/test/test_types_plugin.py +++ b/test/test_types_plugin.py @@ -124,3 +124,10 @@ class TypesPluginTest(unittest.TestCase, TestHelper): def mktime(*args): return time.mktime(datetime(*args).timetuple()) + + +def suite(): + return unittest.TestLoader().loadTestsFromName(__name__) + +if __name__ == '__main__': + unittest.main(defaultTest='suite')