#17333: test_imaplib now works with unittest test discovery. Patch by Zachary Ware.
diff --git a/Lib/test/test_imaplib.py b/Lib/test/test_imaplib.py
index 4b12d23..7db3f7d 100644
--- a/Lib/test/test_imaplib.py
+++ b/Lib/test/test_imaplib.py
@@ -444,7 +444,7 @@
keyfile=CERTFILE, ssl_context=self.create_ssl_context())
-def test_main():
+def load_tests(*args):
tests = [TestImaplib]
if support.is_resource_enabled('network'):
@@ -459,9 +459,9 @@
RemoteIMAPTest, RemoteIMAP_SSLTest, RemoteIMAP_STARTTLSTest,
])
- support.run_unittest(*tests)
+ return unittest.TestSuite([unittest.makeSuite(test) for test in tests])
if __name__ == "__main__":
support.use_resources = ['network']
- test_main()
+ unittest.main()