Andrew Hsieh | ffab958 | 2013-06-18 12:29:14 -0700 | [diff] [blame^] | 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.assertEqual(unicode('abc', codecname), u'abc') |
| 31 | self.assertEqual(unicode('xdef', codecname), u'abcdef') |
| 32 | self.assertEqual(unicode('defx', codecname), u'defabc') |
| 33 | self.assertEqual(unicode('dxf', codecname), u'dabcf') |
| 34 | self.assertEqual(unicode('dxfx', codecname), u'dabcfabc') |
| 35 | |
| 36 | def test_encodex(self): |
| 37 | self.assertEqual(u'abc'.encode(codecname), 'abc') |
| 38 | self.assertEqual(u'xdef'.encode(codecname), 'abcdef') |
| 39 | self.assertEqual(u'defx'.encode(codecname), 'defabc') |
| 40 | self.assertEqual(u'dxf'.encode(codecname), 'dabcf') |
| 41 | self.assertEqual(u'dxfx'.encode(codecname), 'dabcfabc') |
| 42 | |
| 43 | def test_constructory(self): |
| 44 | self.assertEqual(unicode('ydef', codecname), u'def') |
| 45 | self.assertEqual(unicode('defy', codecname), u'def') |
| 46 | self.assertEqual(unicode('dyf', codecname), u'df') |
| 47 | self.assertEqual(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() |