From: Marc-André Lemburg <mal@egenix.com>
Date: Tue, 19 Jun 2001 20:09:28 +0000 (+0000)
Subject: Test by Martin v. Loewis for the new UTF-16 codec handling of BOM
X-Git-Tag: v2.2a3~1512
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=a37171dd8632b80af5500939791e39a482efe4a3;p=thirdparty%2FPython%2Fcpython.git

Test by Martin v. Loewis for the new UTF-16 codec handling of BOM
marks.
---

diff --git a/Lib/test/test_codecs.py b/Lib/test/test_codecs.py
new file mode 100644
index 000000000000..6d9e31d67270
--- /dev/null
+++ b/Lib/test/test_codecs.py
@@ -0,0 +1,25 @@
+import test_support,unittest
+import codecs
+import StringIO
+
+class UTF16Test(unittest.TestCase):
+
+    spamle = '\xff\xfes\x00p\x00a\x00m\x00s\x00p\x00a\x00m\x00'
+    spambe = '\xfe\xff\x00s\x00p\x00a\x00m\x00s\x00p\x00a\x00m'
+
+    def test_only_one_bom(self):
+        _,_,reader,writer = codecs.lookup("utf-16")
+        # encode some stream
+        s = StringIO.StringIO()
+        f = writer(s)
+        f.write(u"spam")
+        f.write(u"spam")
+        d = s.getvalue()
+        # check whether there is exactly one BOM in it
+        self.assert_(d == self.spamle or d == self.spambe)
+        # try to read it back
+        s = StringIO.StringIO(d)
+        f = reader(s)
+        self.assertEquals(f.read(), u"spamspam")
+
+test_support.run_unittest(UTF16Test)