From a74d675facda18cca2a821be56d49446184b6ea1 Mon Sep 17 00:00:00 2001
From: Benjamin Peterson <benjamin@python.org>
Date: Tue, 23 Sep 2008 20:43:09 +0000
Subject: [PATCH] backport the atexit test for r66563

---
 Lib/test/test_atexit.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/Lib/test/test_atexit.py b/Lib/test/test_atexit.py
index bea482f9dd0..a60dfa85b6b 100644
--- a/Lib/test/test_atexit.py
+++ b/Lib/test/test_atexit.py
@@ -22,6 +22,19 @@ class TestCase(unittest.TestCase):
             atexit._exithandlers = save_handlers
         self.assertEqual(s.getvalue(), "h4 (4,) {'kw': 'abc'}\nh4 () {}\nh1\n")
 
+    def test_badargs(self):
+        s = StringIO.StringIO()
+        sys.stdout = sys.stderr = s
+        save_handlers = atexit._exithandlers
+        atexit._exithandlers = []
+        try:
+            atexit.register(lambda: 1, 0, 0, (x for x in (1,2)), 0, 0)
+            self.assertRaises(TypeError, atexit._run_exitfuncs)
+        finally:
+            sys.stdout = sys.__stdout__
+            sys.stderr = sys.__stderr__
+            atexit._exithandlers = save_handlers
+
     def test_order(self):
         # be sure handlers are executed in reverse order
         s = StringIO.StringIO()