fix CC

src/cryptoconditions/tests/test_ed25519.py

@@ -1,6 +1,6 @@
 import json
 import base64
-from test_vectors import jsonRPC
+from .test_vectors import jsonRPC
 
 
 def test_sign_ed25519_pass_simple():

src/cryptoconditions/tests/test_failure_modes.py

@@ -1,7 +1,7 @@
 import json
 import ctypes
 import base64
-from test_vectors import jsonRPC, so, decode_base64 as d64
+from .test_vectors import jsonRPC, so, decode_base64 as d64
 
 
 '''

src/cryptoconditions/tests/test_secp256k1.py

@@ -2,7 +2,7 @@ import json
 import base64
 import hashlib
 import secp256k1
-from test_vectors import jsonRPC, encode_base64, decode_base64
+from .test_vectors import jsonRPC, encode_base64, decode_base64
 
 
 key = secp256k1.PrivateKey()
@@ -20,7 +20,7 @@ def test_sign_secp256k1_pass_simple():
         'message': msg,
     })
 
-    sig = encode_base64(key.ecdsa_serialize_compact(key.ecdsa_sign(msg)))
+    sig = encode_base64(key.ecdsa_serialize_compact(key.ecdsa_sign(msg.encode())))
 
     assert res == {
         "num_signed": 1,

src/cryptoconditions/tests/test_vectors.py

@@ -114,11 +114,15 @@ def decode_base64(data):
 
 
 def encode_base64(data):
-    return base64.urlsafe_b64encode(data).rstrip(b'=')
+    if type(data) == str:
+        data = data.encode()
+    return base64.urlsafe_b64encode(data).rstrip(b'=').decode()
 
 
 def b16_to_b64(b16):
-    return encode_base64(base64.b16decode(b16)).decode()
+    if type(b16) == str:
+        b16 = b16.encode()
+    return encode_base64(base64.b16decode(b16))
 
 
 def _read_vectors(name):