Change the PyUnit-based tests to use the test_main() approach. This

allows using the tests with unittest.py as a script.  The tests will
still run when run as a script themselves.

1 files changed, 7 insertions, 1 deletions

diff --git a/Lib/test/test_iter.py b/Lib/test/test_iter.py
index 37fab7c..f6084cb 100644
--- a/Lib/test/test_iter.py
+++ b/Lib/test/test_iter.py
@@ -694,4 +694,10 @@ class TestCase(unittest.TestCase):
         (a, b), (c,) = IteratingSequenceClass(2), {42: 24}
         self.assertEqual((a, b, c), (0, 1, 42))
 
-run_unittest(TestCase)
+
+def test_main():
+    run_unittest(TestCase)
+
+
+if __name__ == "__main__":
+    test_main()