]> granicus.if.org Git - python/commitdiff
allow more exceptions
authorBenjamin Peterson <benjamin@python.org>
Fri, 9 Jul 2010 13:28:42 +0000 (13:28 +0000)
committerBenjamin Peterson <benjamin@python.org>
Fri, 9 Jul 2010 13:28:42 +0000 (13:28 +0000)
Lib/test/test_struct.py

index 23ec29afddd321150560174439cfe8f03d663e33..96efa7350a5ff91969d4d4b5164d6a77834b94fc 100644 (file)
@@ -47,8 +47,8 @@ class StructTest(unittest.TestCase):
 
         self.assertRaises(struct.error, struct.pack, 'iii', 3)
         self.assertRaises(struct.error, struct.pack, 'i', 3, 3, 3)
-        self.assertRaises(struct.error, struct.pack, 'i', 'foo')
-        self.assertRaises(struct.error, struct.pack, 'P', 'foo')
+        self.assertRaises((TypeError, struct.error), struct.pack, 'i', 'foo')
+        self.assertRaises((TypeError, struct.error), struct.pack, 'P', 'foo')
         self.assertRaises(struct.error, struct.unpack, 'd', b'flap')
         s = struct.pack('ii', 1, 2)
         self.assertRaises(struct.error, struct.unpack, 'iii', s)
@@ -446,7 +446,7 @@ class StructTest(unittest.TestCase):
 
         # Test bogus offset (issue 3694)
         sb = small_buf
-        self.assertRaises(TypeError, struct.pack_into, b'', sb, None)
+        self.assertRaises((TypeError, struct.error), struct.pack_into, b'', sb, None)
 
     def test_pack_into_fn(self):
         test_string = b'Reykjavik rocks, eow!'