Support multiple test modules.

dkim/tests/__init__.py

@@ -0,0 +1,30 @@
+# This software is provided 'as-is', without any express or implied
+# warranty.  In no event will the author be held liable for any damages
+# arising from the use of this software.
+#
+# Permission is granted to anyone to use this software for any purpose,
+# including commercial applications, and to alter it and redistribute it
+# freely, subject to the following restrictions:
+#
+# 1. The origin of this software must not be misrepresented; you must not
+#    claim that you wrote the original software. If you use this software
+#    in a product, an acknowledgment in the product documentation would be
+#    appreciated but is not required.
+# 2. Altered source versions must be plainly marked as such, and must not be
+#    misrepresented as being the original software.
+# 3. This notice may not be removed or altered from any source distribution.
+#
+# Copyright (c) 2011 William Grant <me@williamgrant.id.au>
+
+import unittest
+
+
+def test_suite():
+    from dkim.tests import (
+        test_dkim,
+        )
+    modules = [
+        test_dkim,
+        ]
+    suites = map(lambda x: x.test_suite(), modules)
+    return unittest.TestSuite(suites)

dkim/tests/test_dkim.py

@@ -112,5 +112,6 @@ class TestParseTagValue(unittest.TestCase):
             parse_tag_value, 'foo=bar;foo=baz')
 
 
-if __name__ == '__main__':
-    unittest.main()
+def test_suite():
+    from unittest import TestLoader
+    return TestLoader().loadTestsFromName(__name__)