arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-31 10:30:58 +00:00
Code Issues Releases Wiki Activity

device_tests: update test_msg_getentropy to new style

Browse Source
This commit is contained in:
matejcik 2018-08-13 15:38:04 +02:00
parent ae5341c1bd
commit e336f578af
1 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
trezorlib/tests/device_tests/test_msg_getentropy.py
View File

@ -15,32 +15,34 @@
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
import math
from .common import TrezorTest
from trezorlib import misc
import trezorlib.messages as proto
import pytest
from trezorlib import messages as m, misc
ENTROPY_LENGTHS_POW2 = [2 ** l for l in range(10)]
ENTROPY_LENGTHS_POW2_1 = [2 ** l + 1 for l in range(10)]
ENTROPY_LENGTHS = ENTROPY_LENGTHS_POW2 + ENTROPY_LENGTHS_POW2_1
def entropy(data):
counts = {}
for c in data:
if c in counts:
counts[c] += 1
else:
counts[c] = 1
counts[c] = counts.get(c, 0) + 1
e = 0
for _, v in counts.items():
p = 1.0 * v / len(data)
for v in counts.values():
p = v / len(data)
e -= p * math.log(p, 256)
return e
class TestMsgGetentropy(TrezorTest):
def test_entropy(self):
for l in [0, 1, 2, 3, 4, 5, 8, 9, 16, 17, 32, 33, 64, 65, 128, 129, 256, 257, 512, 513, 1024]:
with self.client:
self.client.set_expected_responses([proto.ButtonRequest(code=proto.ButtonRequestType.ProtectCall), proto.Entropy()])
ent = misc.get_entropy(self.client, l)
assert len(ent) == l
print('entropy = ', entropy(ent))
@pytest.mark.parametrize("entropy_length", ENTROPY_LENGTHS)
def test_entropy(client, entropy_length):
with client:
client.set_expected_responses(
[m.ButtonRequest(code=m.ButtonRequestType.ProtectCall), m.Entropy()]
)
ent = misc.get_entropy(client, entropy_length)
assert len(ent) == entropy_length
print("{} bytes: entropy = {}".format(entropy_length, entropy(ent)))