diff Lib/test/test_codecs.py @ 61850:846dbcf970ab 3.1

Merged revisions 81461 via svnmerge from svn+ssh://pythondev@svn.python.org/python/branches/py3k ................ r81461 | victor.stinner | 2010-05-22 04:16:27 +0200 (sam., 22 mai 2010) | 10 lines Merged revisions 81459 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r81459 | victor.stinner | 2010-05-22 04:11:07 +0200 (sam., 22 mai 2010) | 3 lines Issue #6268: Fix seek() method of codecs.open(), don't read the BOM twice after seek(0) ........ ................
author Victor Stinner <victor.stinner@haypocalc.com>
date Sat, 22 May 2010 02:17:42 +0000
parents 149badf73bca
children cd4936691163
line wrap: on
line diff
--- a/Lib/test/test_codecs.py	Fri May 21 22:05:15 2010 +0000
+++ b/Lib/test/test_codecs.py	Sat May 22 02:17:42 2010 +0000
@@ -1592,6 +1592,26 @@
                          b"\xe4\xeb\xef\xf6\xfc")
 
 
+class BomTest(unittest.TestCase):
+    def test_seek0(self):
+        data = "1234567890"
+        tests = ("utf-16",
+                 "utf-16-le",
+                 "utf-16-be",
+                 "utf-32",
+                 "utf-32-le",
+                 "utf-32-be")
+        for encoding in tests:
+            with codecs.open('foo', 'w+', encoding=encoding) as f:
+                # Check if the BOM is written only once
+                f.write(data)
+                f.write(data)
+                f.seek(0)
+                self.assertEquals(f.read(), data * 2)
+                f.seek(0)
+                self.assertEquals(f.read(), data * 2)
+
+
 def test_main():
     support.run_unittest(
         UTF32Test,
@@ -1619,6 +1639,7 @@
         WithStmtTest,
         TypesTest,
         SurrogateEscapeTest,
+        BomTest,
     )