arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-06-28 02:42:34 +00:00
Code Issues Releases Wiki Activity

python3 migration (#149)

Browse Source
This commit is contained in:
onvej-sl 2018-04-05 16:02:07 +02:00 committed by Pavol Rusnak
parent 7ea31aecf7
commit 8b61a7e365
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_curves.py
View File

@ -86,7 +86,7 @@ def r(request):
@pytest.fixture(params=list(sorted(curves))) @pytest.fixture(params=list(sorted(curves)))
def curve(request): def curve(request):
name = request.param name = request.param
curve_ptr = lib.get_curve_by_name(name).contents.params curve_ptr = lib.get_curve_by_name(bytes(name, "ascii")).contents.params
assert curve_ptr, 'curve {} not found'.format(name) assert curve_ptr, 'curve {} not found'.format(name)
curve_obj = curves[name] curve_obj = curves[name]
curve_obj.ptr = c.c_void_p(curve_ptr) curve_obj.ptr = c.c_void_p(curve_ptr)
@ -96,7 +96,7 @@ def curve(request):
@pytest.fixture(params=points) @pytest.fixture(params=points)
def point(request): def point(request):
name = request.param.curve name = request.param.curve
curve_ptr = lib.get_curve_by_name(name).contents.params curve_ptr = lib.get_curve_by_name(bytes(name, "ascii")).contents.params
assert curve_ptr, 'curve {} not found'.format(name) assert curve_ptr, 'curve {} not found'.format(name)
curve_obj = curves[name] curve_obj = curves[name]
curve_obj.ptr = c.c_void_p(curve_ptr) curve_obj.ptr = c.c_void_p(curve_ptr)
@ -435,4 +435,4 @@ def test_curve25519_scalarmult_from_gpg(r):
res = r.randbytes(32) res = r.randbytes(32)
lib.curve25519_scalarmult(res, sec[::-1], pub[::-1]) lib.curve25519_scalarmult(res, sec[::-1], pub[::-1])
expected = 'a93dbdb23e5c99da743e203bd391af79f2b83fb8d0fd6ec813371c71f08f2d4d' expected = 'a93dbdb23e5c99da743e203bd391af79f2b83fb8d0fd6ec813371c71f08f2d4d'
assert binascii.hexlify(bytearray(res)) == expected assert binascii.hexlify(bytearray(res)) == bytes(expected, "ascii")