* fix the print test

* add more __init__ tests

diff --git a/Lib/test/test_set.py b/Lib/test/test_set.py
index 514b75c0d69cfaec4abe3c91fc2e5f78a8965b54..a4cbfc5ade7e0aaf4146790b28f3dafb78d8f59c 100644 (file)
--- a/Lib/test/test_set.py
+++ b/Lib/test/test_set.py
@@ -4,6 +4,7 @@ from weakref import proxy
 import operator
 import copy
 import pickle
+import os
 
 class PassThru(Exception):
     pass
@@ -201,6 +202,8 @@ class TestSet(TestJointOps):
         self.assertEqual(s, set(self.word))
         s.__init__(self.otherword)
         self.assertEqual(s, set(self.otherword))
+        self.assertRaises(TypeError, s.__init__, s, 2);
+        self.assertRaises(TypeError, s.__init__, 1);
 
     def test_constructor_identity(self):
         s = self.thetype(range(3))
@@ -436,6 +439,17 @@ class TestBasicOps(unittest.TestCase):
         if self.repr is not None:
             self.assertEqual(repr(self.set), self.repr)
 
+    def test_print(self):
+        try:
+            fo = open(test_support.TESTFN, "wb")
+            print >> fo, self.set,
+            fo.close()
+            fo = open(test_support.TESTFN, "rb")
+            self.assertEqual(fo.read(), repr(self.set))
+        finally:
+            fo.close()
+            os.remove(test_support.TESTFN)
+
     def test_length(self):
         self.assertEqual(len(self.set), self.length)