From 61175c969b1de3ecd8c25478c69d6d1883dfa211 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Damir=20Jeli=C4=87?= <poljar@termina.org.uk>
Date: Thu, 20 Jun 2019 14:05:36 +0200
Subject: tests: Simplify the input parameter for the Unicode decoding tests.

---
 python/tests/group_session_test.py | 5 +----
 python/tests/pk_test.py            | 4 +---
 python/tests/session_test.py       | 4 +---
 3 files changed, 3 insertions(+), 10 deletions(-)

(limited to 'python/tests')

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)
-- 
cgit v1.2.3