Removed compatibility code for Python 2.7 and 3.4
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 7ce57eb..7cf7ed4 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -15,21 +15,15 @@
import rsa
import rsa.cli
import rsa.util
-from rsa._compat import PY2
-def make_buffer():
- if PY2:
- return BytesIO()
+def make_buffer() -> StringIO:
buf = StringIO()
buf.buffer = BytesIO()
return buf
-def get_bytes_out(out):
- if PY2:
- # Python 2.x writes 'str' to stdout
- return out.getvalue()
+def get_bytes_out(out: StringIO) -> bytes:
# Python 3.x writes 'bytes' to stdout.buffer
return out.buffer.getvalue()