Lib/test/test_charmapcodec.py

   1 """ Python character mapping codec test
   2
   3 This uses the test codec in testcodec.py and thus also tests the
   4 encodings package lookup scheme.
   5
   6 Written by Marc-Andre Lemburg (mal@lemburg.com).
   7
   8 (c) Copyright 2000 Guido van Rossum.
   9
  10 """#"
  11
  12 import test.test_support, unittest
  13
  14 import codecs
  15
  16 # Register a search function which knows about our codec
  17 def codec_search_function(encoding):
  18     if encoding == 'testcodec':
  19         from test import testcodec
  20         return tuple(testcodec.getregentry())
  21     return None
  22
  23 codecs.register(codec_search_function)
  24
  25 # test codec's name (see test/testcodec.py)
  26 codecname = 'testcodec'
  27
  28 class CharmapCodecTest(unittest.TestCase):
  29     def test_constructorx(self):
  30         self.assertEquals(unicode('abc', codecname), u'abc')
  31         self.assertEquals(unicode('xdef', codecname), u'abcdef')
  32         self.assertEquals(unicode('defx', codecname), u'defabc')
  33         self.assertEquals(unicode('dxf', codecname), u'dabcf')
  34         self.assertEquals(unicode('dxfx', codecname), u'dabcfabc')
  35
  36     def test_encodex(self):
  37         self.assertEquals(u'abc'.encode(codecname), 'abc')
  38         self.assertEquals(u'xdef'.encode(codecname), 'abcdef')
  39         self.assertEquals(u'defx'.encode(codecname), 'defabc')
  40         self.assertEquals(u'dxf'.encode(codecname), 'dabcf')
  41         self.assertEquals(u'dxfx'.encode(codecname), 'dabcfabc')
  42
  43     def test_constructory(self):
  44         self.assertEquals(unicode('ydef', codecname), u'def')
  45         self.assertEquals(unicode('defy', codecname), u'def')
  46         self.assertEquals(unicode('dyf', codecname), u'df')
  47         self.assertEquals(unicode('dyfy', codecname), u'df')
  48
  49     def test_maptoundefined(self):
  50         self.assertRaises(UnicodeError, unicode, 'abc\001', codecname)
  51
  52 def test_main():
  53     test.test_support.run_unittest(CharmapCodecTest)
  54
  55 if __name__ == "__main__":
  56     test_main()