#17479: test_io now works with unittest test discovery. Patch by Zachary Ware.
diff --git a/Lib/test/test_io.py b/Lib/test/test_io.py
index 64c66d8..95b4d80 100644
--- a/Lib/test/test_io.py
+++ b/Lib/test/test_io.py
@@ -3226,7 +3226,7 @@
test_reentrant_write_text = None
-def test_main():
+def load_tests(*args):
tests = (CIOTest, PyIOTest,
CBufferedReaderTest, PyBufferedReaderTest,
CBufferedWriterTest, PyBufferedWriterTest,
@@ -3259,7 +3259,8 @@
for name, obj in py_io_ns.items():
setattr(test, name, obj)
- support.run_unittest(*tests)
+ suite = unittest.TestSuite([unittest.makeSuite(test) for test in tests])
+ return suite
if __name__ == "__main__":
- test_main()
+ unittest.main()
diff --git a/Misc/NEWS b/Misc/NEWS
index 0a94579..9a0c9dd 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -714,6 +714,9 @@
- Issue #15539: Added regression tests for Tools/scripts/pindent.py.
+- Issue #17479: test_io now works with unittest test discovery.
+ Patch by Zachary Ware.
+
- Issue #17066: test_robotparser now works with unittest test discovery.
Patch by Zachary Ware.