Merge pull request #6 manually

Improve error message when the type is not bytes
keis · Jun 14, 2016 · 3f9b28c · 3f9b28c
2 parents 600969b + ffbb84f
commit 3f9b28c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/base58.py b/base58.py
@@ -30,6 +30,10 @@
 def b58encode(v):
     '''Encode a string using Base58'''
 
+    if not isinstance(v, bytes):
+        raise TypeError("a bytes-like object is required, not '%s'" %
+                        type(v).__name__)
+
     origlen = len(v)
     v = v.lstrip(b'\0')
     newlen = len(v)

diff --git a/test.py b/test.py
@@ -77,3 +77,9 @@ def test_round_trips():
             bytes_in = b''.join(bytes_to_test)
             bytes_out = b58decode(b58encode(bytes_in))
             assert_that(bytes_in, equal_to(bytes_out))
+
+
+def test_input_should_be_bytes():
+     data = u'3vQB7B6MrGQZaxCuFg4oH'
+     with assert_raises(TypeError):
+         b58encode(data)