diff options
-rw-r--r-- | python/tests/group_session_test.py | 5 | ||||
-rw-r--r-- | python/tests/pk_test.py | 4 | ||||
-rw-r--r-- | python/tests/session_test.py | 4 |
3 files changed, 3 insertions, 10 deletions
diff --git a/python/tests/group_session_test.py b/python/tests/group_session_test.py index b45fbb6..4632a60 100644 --- a/python/tests/group_session_test.py +++ b/python/tests/group_session_test.py @@ -1,7 +1,4 @@ # -*- coding: utf-8 -*- - -from builtins import bytes - import pytest from olm import InboundGroupSession, OlmGroupSessionError, OutboundGroupSession @@ -121,7 +118,7 @@ class TestClass(object): outbound = OutboundGroupSession() inbound = InboundGroupSession(outbound.session_key) - text = outbound.encrypt(bytes([0xed])) + text = outbound.encrypt(b"\xed") plaintext, _ = inbound.decrypt(text) print(plaintext) diff --git a/python/tests/pk_test.py b/python/tests/pk_test.py index b82f420..ef87465 100644 --- a/python/tests/pk_test.py +++ b/python/tests/pk_test.py @@ -1,6 +1,4 @@ # -*- coding: utf-8 -*- -from builtins import bytes - import pytest from olm import (PkDecryption, PkDecryptionError, PkEncryption, PkSigning, @@ -62,6 +60,6 @@ class TestClass(object): def test_invalid_unicode_decrypt(self): decryption = PkDecryption() encryption = PkEncryption(decryption.public_key) - message = encryption.encrypt(bytes([0xed])) + message = encryption.encrypt(b"\xed") plaintext = decryption.decrypt(message) assert plaintext == u"�" diff --git a/python/tests/session_test.py b/python/tests/session_test.py index dce37c5..b856585 100644 --- a/python/tests/session_test.py +++ b/python/tests/session_test.py @@ -1,6 +1,4 @@ # -*- coding: utf-8 -*- -from builtins import bytes - import pytest from olm import (Account, InboundSession, OlmMessage, OlmPreKeyMessage, @@ -147,7 +145,7 @@ class TestClass(object): def test_invalid_unicode_decrypt(self): alice, bob, session = self._create_session() - message = session.encrypt(bytes([0xed])) + message = session.encrypt(b"\xed") bob_session = InboundSession(bob, message) plaintext = bob_session.decrypt(message) |