Drop byte_literal in favour of b''
diff --git a/tests/test_cli.py b/tests/test_cli.py
index aad734a..4ae8da3 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -15,7 +15,6 @@
import rsa
import rsa.cli
import rsa.util
-from rsa._compat import b
if sys.version_info[0] < 3:
def make_buffer():
@@ -135,8 +134,8 @@
rsa.cli.keygen()
lines = get_bytes_out(out).splitlines()
- self.assertEqual(b('-----BEGIN RSA PRIVATE KEY-----'), lines[0])
- self.assertEqual(b('-----END RSA PRIVATE KEY-----'), lines[-1])
+ self.assertEqual(b'-----BEGIN RSA PRIVATE KEY-----', lines[0])
+ self.assertEqual(b'-----END RSA PRIVATE KEY-----', lines[-1])
# The key size should be shown on stderr
self.assertTrue('128-bit key' in err.getvalue())
@@ -217,7 +216,7 @@
# We should have the original cleartext on stdout now.
output = get_bytes_out(out)
- self.assertEqual(b('Hello cleartext RSA users!'), output)
+ self.assertEqual(b'Hello cleartext RSA users!', output)
@cleanup_files('encrypted.txt', 'cleartext.txt')
def test_encrypt_decrypt_unhappy(self):