forked from keis/base58
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.py
59 lines (39 loc) · 1.4 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from contextlib import contextmanager
from hamcrest import assert_that, equal_to, instance_of
from base58 import b58encode, b58decode, b58encode_check, b58decode_check
class RaisesContext(object):
pass
@contextmanager
def assert_raises(matcher=None, message=''):
# Short hand for instance_of matcher
if isinstance(matcher, (type,)):
matcher = instance_of(matcher)
context = RaisesContext()
try:
yield context
except Exception as e:
context.exception = e
assert_that(context.exception, matcher, message)
def test_simple_encode():
data = b58encode(b'hello world')
assert_that(data, equal_to('StV1DL6CwTryKyV'))
def test_leadingz_encode():
data = b58encode(b'\0\0hello world')
assert_that(data, equal_to('11StV1DL6CwTryKyV'))
def test_simple_decode():
data = b58decode('StV1DL6CwTryKyV')
assert_that(data, equal_to(b'hello world'))
def test_simple_decode_bytes():
data = b58decode(b'StV1DL6CwTryKyV')
assert_that(data, equal_to(b'hello world'))
def test_leadingz_decode():
data = b58decode('11StV1DL6CwTryKyV')
assert_that(data, equal_to(b'\0\0hello world'))
def test_check_identity():
data = b'hello world'
out = b58decode_check(b58encode_check(data))
assert_that(out, equal_to(data))
def test_check_failure():
data = '3vQB7B6MrGQZaxCuFg4oH'
with assert_raises(ValueError):
b58decode_check(data)