Mercurial > cpython-lru_cache-weakref

diff Lib/test/test_bytes.py @ 44137:825cffd6c0cd

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Issue #1283: Allow any iterable of integers to be passed to bytearray.extend().
author Alexandre Vassalotti <alexandre@peadrop.com>
date Tue, 04 Dec 2007 05:51:13 +0000
parents 14c4bb3c65a7
children 9d152530910a
line wrap: on
line diff
--- a/Lib/test/test_bytes.py	Tue Dec 04 01:13:14 2007 +0000
+++ b/Lib/test/test_bytes.py	Tue Dec 04 05:51:13 2007 +0000
@@ -529,6 +529,24 @@
         a.extend(a)
         self.assertEqual(a, orig + orig)
         self.assertEqual(a[5:], orig)
+        a = bytearray(b'')
+        # Test iterators that don't have a __length_hint__
+        a.extend(map(int, orig * 25))
+        a.extend(int(x) for x in orig * 25)
+        self.assertEqual(a, orig * 50)
+        self.assertEqual(a[-5:], orig)
+        a = bytearray(b'')
+        a.extend(iter(map(int, orig * 50)))
+        self.assertEqual(a, orig * 50)
+        self.assertEqual(a[-5:], orig)
+        a = bytearray(b'')
+        a.extend(list(map(int, orig * 50)))
+        self.assertEqual(a, orig * 50)
+        self.assertEqual(a[-5:], orig)
+        a = bytearray(b'')
+        self.assertRaises(ValueError, a.extend, [0, 1, 2, 256])
+        self.assertRaises(ValueError, a.extend, [0, 1, 2, -1])
+        self.assertEqual(len(a), 0)
 
     def test_remove(self):
         b = bytearray(b'hello')