diff options
Diffstat (limited to 'tests/test_transform.py')
| -rw-r--r-- | tests/test_transform.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tests/test_transform.py b/tests/test_transform.py index ecc1a30..9bd3c6d 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -3,7 +3,7 @@ import unittest2 from rsa._compat import b -from rsa.transform import int2bytes +from rsa.transform import int2bytes, old_int2bytes class Test_integer_to_bytes(unittest2.TestCase): @@ -12,13 +12,21 @@ class Test_integer_to_bytes(unittest2.TestCase): b('\x00\x00\x07[\xcd\x15')) self.assertEqual(int2bytes(123456789, 7), b('\x00\x00\x00\x07[\xcd\x15')) + self.assertEqual(old_int2bytes(123456789, 6), + b('\x00\x00\x07[\xcd\x15')) + self.assertEqual(old_int2bytes(123456789, 7), + b('\x00\x00\x00\x07[\xcd\x15')) def test_raises_OverflowError_when_chunk_size_is_insufficient(self): self.assertRaises(OverflowError, int2bytes, 123456789, 3) self.assertRaises(OverflowError, int2bytes, 299999999999, 4) + self.assertRaises(OverflowError, old_int2bytes, 123456789, 3) + self.assertRaises(OverflowError, old_int2bytes, 299999999999, 4) def test_raises_ValueError_when_negative_integer(self): self.assertRaises(ValueError, int2bytes, -1) + self.assertRaises(ValueError, old_int2bytes, -1) def test_raises_TypeError_when_not_integer(self): self.assertRaises(TypeError, int2bytes, None) + self.assertRaises(TypeError, old_int2bytes, None) |