zec
/
zcash-test-vectors
mirror of https://github.com/zcash/zcash-test-vectors.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Render Vec<u8> data correctly

This commit is contained in:
Jack Grigg 2018-08-21 19:44:02 +01:00
parent af98406307
commit a9676cc9b4
No known key found for this signature in database
GPG Key ID: 1B8D649257DB0829
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tv_output.py
View File

@ -63,6 +63,17 @@ def tv_bytes_rust(name, value, pad):
pad,
))
def tv_vec_bytes_rust(name, value, pad):
print('''%s%s: vec![
%s%s
%s],''' % (
pad,
name,
pad,
chunk(hexlify(value)),
pad,
))
def tv_option_bytes_rust(name, value, pad):
if value:
print('''%s%s: Some([
@ -80,10 +91,13 @@ def tv_option_bytes_rust(name, value, pad):
def tv_int_rust(name, value, pad):
print('%s%s: %d,' % (pad, name, value))
def tv_part_rust(name, value, indent=3):
def tv_part_rust(name, value, typ, indent=3):
pad = ' ' * indent
if type(value) == bytes:
tv_bytes_rust(name, value, pad)
if typ == 'Vec<u8>':
tv_vec_bytes_rust(name, value, pad)
else:
tv_bytes_rust(name, value, pad)
elif isinstance(value, Some) or value is None:
tv_option_bytes_rust(name, value, pad)
elif type(value) == int:
@ -101,13 +115,13 @@ def tv_rust(filename, parts, vectors):
))
if type(vectors) == type({}):
print(' let test_vector = TestVector {')
for p in parts: tv_part_rust(p[0], vectors[p[0]])
for p in parts: tv_part_rust(p[0], vectors[p[0]], p[1]['rust'])
print(' };')
elif type(vectors) == type([]):
print(' let test_vectors = vec![')
for vector in vectors:
print(' TestVector {')
for p in parts: tv_part_rust(p[0], vector[p[0]], 4)
for p in parts: tv_part_rust(p[0], vector[p[0]], p[1]['rust'], 4)
print(' },')
print(' ];')
else: