From 1eb1fb814b4d5ea12dcffd18bb132a1313a48ccf Mon Sep 17 00:00:00 2001
From: Raymond Hettinger <python@rcn.com>
Date: Fri, 8 Nov 2002 05:03:21 +0000
Subject: [PATCH] Closes SF bug #628246.

The _update method detected mutable elements by trapping TypeErrors.
Unfortunately, this masked useful TypeErrors raised by the iterable
itself.  For cases where it is possible for an iterable to raise
a TypeError, the iterable is pre-converted to a list outside the
try/except so that any TypeErrors propagate through.
---
 Lib/sets.py           |  2 ++
 Lib/test/test_sets.py | 25 +++++++++++++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/Lib/sets.py b/Lib/sets.py
index 5f0f0a2d38..bbb93a091e 100644
--- a/Lib/sets.py
+++ b/Lib/sets.py
@@ -320,6 +320,8 @@ class BaseSet(object):
             return
 
         value = True
+        if type(iterable) not in (list, tuple, dict, file, xrange, str):
+            iterable = list(iterable)
         it = iter(iterable)
         while True:
             try:
diff --git a/Lib/test/test_sets.py b/Lib/test/test_sets.py
index 840036cc65..76b56b118d 100644
--- a/Lib/test/test_sets.py
+++ b/Lib/test/test_sets.py
@@ -132,6 +132,30 @@ class TestBasicOpsTriple(TestBasicOps):
 
 #==============================================================================
 
+def baditer():
+    raise TypeError
+    yield True
+
+def gooditer():
+    yield True
+
+class TestExceptionPropagation(unittest.TestCase):
+    """SF 628246:  Set constructor should not trap iterator TypeErrors"""
+
+    def test_instanceWithException(self):
+        self.assertRaises(TypeError, Set, baditer())
+
+    def test_instancesWithoutException(self):
+        """All of these iterables should load without exception."""
+        Set([1,2,3])
+        Set((1,2,3))
+        Set({'one':1, 'two':2, 'three':3})
+        Set(xrange(3))
+        Set('abc')
+        Set(gooditer())
+
+#==============================================================================
+
 class TestSetOfSets(unittest.TestCase):
     def test_constructor(self):
         inner = Set([1])
@@ -604,6 +628,7 @@ class TestCopyingNested(TestCopying):
 def makeAllTests():
     suite = unittest.TestSuite()
     for klass in (TestSetOfSets,
+                  TestExceptionPropagation,
                   TestBasicOpsEmpty,
                   TestBasicOpsSingleton,
                   TestBasicOpsTuple,
-- 
2.49.0