| """ 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() |