diff options
Diffstat (limited to 'tests/test_keys.py')
-rw-r--r-- | tests/test_keys.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_keys.py b/tests/test_keys.py index cfdfa0b..eefd48f 100644 --- a/tests/test_keys.py +++ b/tests/test_keys.py @@ -13,9 +13,9 @@ from ._unittest_compat import patch patch() if sys.version_info < (3,): - num_cls = long # noqa + int_types = (int, long) # noqa else: - num_cls = int + int_types = int tests_root = os.path.dirname(__file__) fixtures_dir = os.path.join(tests_root, 'fixtures') @@ -487,9 +487,9 @@ class KeysTests(unittest.TestCase): with open(os.path.join(fixtures_dir, public_key_file), 'rb') as f: public_key = keys.PublicKeyInfo.load(f.read()) - self.assertIsInstance(private_key.bit_size, num_cls) + self.assertIsInstance(private_key.bit_size, int_types) self.assertEqual(bit_size, private_key.bit_size) - self.assertIsInstance(public_key.bit_size, num_cls) + self.assertIsInstance(public_key.bit_size, int_types) self.assertEqual(bit_size, public_key.bit_size) @staticmethod |