summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_compat.py2
-rw-r--r--tests/test_prime.py1
2 files changed, 2 insertions, 1 deletions
diff --git a/tests/test_compat.py b/tests/test_compat.py
index 74d6f53..2a58df5 100644
--- a/tests/test_compat.py
+++ b/tests/test_compat.py
@@ -18,7 +18,7 @@ import unittest
import struct
import sys
-from rsa._compat import is_bytes, byte, b
+from rsa._compat import b, byte, is_bytes, range
class TestByte(unittest.TestCase):
diff --git a/tests/test_prime.py b/tests/test_prime.py
index 75b80b3..f3bda9b 100644
--- a/tests/test_prime.py
+++ b/tests/test_prime.py
@@ -18,6 +18,7 @@
import unittest
+from rsa._compat import range
import rsa.prime
import rsa.randnum