diff options
Diffstat (limited to 'tests/test_transform.py')
| -rw-r--r-- | tests/test_transform.py | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/tests/test_transform.py b/tests/test_transform.py index 7b9335e..1404619 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -18,20 +18,19 @@ from rsa.transform import int2bytes, bytes2int class Test_int2bytes(unittest.TestCase): def test_accuracy(self): - self.assertEqual(int2bytes(123456789), b'\x07[\xcd\x15') + self.assertEqual(int2bytes(123456789), b"\x07[\xcd\x15") def test_codec_identity(self): self.assertEqual(bytes2int(int2bytes(123456789, 128)), 123456789) def test_chunk_size(self): - self.assertEqual(int2bytes(123456789, 6), b'\x00\x00\x07[\xcd\x15') - self.assertEqual(int2bytes(123456789, 7), - b'\x00\x00\x00\x07[\xcd\x15') + self.assertEqual(int2bytes(123456789, 6), b"\x00\x00\x07[\xcd\x15") + self.assertEqual(int2bytes(123456789, 7), b"\x00\x00\x00\x07[\xcd\x15") def test_zero(self): - self.assertEqual(int2bytes(0, 4), b'\x00' * 4) - self.assertEqual(int2bytes(0, 7), b'\x00' * 7) - self.assertEqual(int2bytes(0), b'\x00') + self.assertEqual(int2bytes(0, 4), b"\x00" * 4) + self.assertEqual(int2bytes(0, 7), b"\x00" * 7) + self.assertEqual(int2bytes(0), b"\x00") def test_correctness_against_base_implementation(self): # Slow test. @@ -41,9 +40,7 @@ class Test_int2bytes(unittest.TestCase): 1 << 77, ] for value in values: - self.assertEqual(bytes2int(int2bytes(value)), - value, - "Boom %d" % value) + self.assertEqual(bytes2int(int2bytes(value)), value, "Boom %d" % value) def test_raises_OverflowError_when_chunk_size_is_insufficient(self): self.assertRaises(OverflowError, int2bytes, 123456789, 3) |