| Yesudeep Mangalapilly | fd1ec36 | 2011-08-11 01:47:46 +0530 | [diff] [blame^] | 1 | # -*- coding: utf-8 -*- |
| 2 | |||||
| 3 | import unittest2 | ||||
| 4 | import struct | ||||
| 5 | |||||
| 6 | from rsa._compat import is_bytes, byte | ||||
| 7 | |||||
| 8 | class Test_byte(unittest2.TestCase): | ||||
| 9 | def test_byte(self): | ||||
| 10 | for i in range(256): | ||||
| 11 | byt = byte(i) | ||||
| 12 | self.assertTrue(is_bytes(byt)) | ||||
| 13 | self.assertEqual(ord(byt), i) | ||||
| 14 | |||||
| 15 | def test_raises_StructError_on_overflow(self): | ||||
| 16 | self.assertRaises(struct.error, byte, 256) | ||||
| 17 | self.assertRaises(struct.error, byte, -1) | ||||