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