cpython/Lib/test/test_charmapcodec.py

57 lines
1.8 KiB
Python

""" Python character mapping codec test
This uses the test codec in testcodec.py and thus also tests the
encodings package lookup scheme.
Written by Marc-Andre Lemburg (mal@lemburg.com).
(c) Copyright 2000 Guido van Rossum.
"""#"
import test.test_support, unittest
import codecs
# Register a search function which knows about our codec
def codec_search_function(encoding):
if encoding == 'testcodec':
from test import testcodec
return tuple(testcodec.getregentry())
return None
codecs.register(codec_search_function)
# test codec's name (see test/testcodec.py)
codecname = 'testcodec'
class CharmapCodecTest(unittest.TestCase):
def test_constructorx(self):
self.assertEqual(unicode('abc', codecname), u'abc')
self.assertEqual(unicode('xdef', codecname), u'abcdef')
self.assertEqual(unicode('defx', codecname), u'defabc')
self.assertEqual(unicode('dxf', codecname), u'dabcf')
self.assertEqual(unicode('dxfx', codecname), u'dabcfabc')
def test_encodex(self):
self.assertEqual(u'abc'.encode(codecname), 'abc')
self.assertEqual(u'xdef'.encode(codecname), 'abcdef')
self.assertEqual(u'defx'.encode(codecname), 'defabc')
self.assertEqual(u'dxf'.encode(codecname), 'dabcf')
self.assertEqual(u'dxfx'.encode(codecname), 'dabcfabc')
def test_constructory(self):
self.assertEqual(unicode('ydef', codecname), u'def')
self.assertEqual(unicode('defy', codecname), u'def')
self.assertEqual(unicode('dyf', codecname), u'df')
self.assertEqual(unicode('dyfy', codecname), u'df')
def test_maptoundefined(self):
self.assertRaises(UnicodeError, unicode, 'abc\001', codecname)
def test_main():
test.test_support.run_unittest(CharmapCodecTest)
if __name__ == "__main__":
test_main()