Fixed test

xxtea · Dec 10, 2018 · 3a6f690 · 3a6f690
1 parent bd11fe0
commit 3a6f690
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/xxtea/__init__.py b/xxtea/__init__.py
@@ -58,8 +58,8 @@ def decrypt_utf8(data, key):
     text = "Hello World! \0你好，中国！"
     key = "1234567890"
     encrypt_data = encrypt(text, key)
-if sys.version_info < (3, 0):
-    decrypt_data = decrypt(encrypt_data, key)
-else:
-    decrypt_data = decrypt_utf8(encrypt_data, key)
+    if sys.version_info < (3, 0):
+        decrypt_data = decrypt(encrypt_data, key)
+    else:
+        decrypt_data = decrypt_utf8(encrypt_data, key)
     assert(text == decrypt_data)