47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
|
import json
|
||
|
|
import re
|
||
|
|
|
||
|
|
output = ""
|
||
|
|
for line in file("test-vectors-00.txt"):
|
||
|
|
line = line.strip()
|
||
|
|
if line == "" or line[0:1] == "#":
|
||
|
|
continue
|
||
|
|
if line.startswith("Josefsson") or line.startswith("Internet-Draft"):
|
||
|
|
continue
|
||
|
|
output += line.replace("\n", "")
|
||
|
|
|
||
|
|
Tests = [ ]
|
||
|
|
|
||
|
|
def get_byte(v):
|
||
|
|
if len(v) == 1:
|
||
|
|
return ord(v)
|
||
|
|
return int(v[2:4], 16)
|
||
|
|
|
||
|
|
def get_string(value):
|
||
|
|
value = value.strip()
|
||
|
|
if value[0] == '"' and value[-1] == '"':
|
||
|
|
return map(get_byte, re.findall("(\\\\x[0-9a-fA-F]{2}|.)", value[1:-1].replace('""', '')))
|
||
|
|
if value.lower() == "null":
|
||
|
|
return None
|
||
|
|
raise Exception("unhandled")
|
||
|
|
|
||
|
|
Tests = [ ]
|
||
|
|
|
||
|
|
matches = re.findall("({(?:.|\n)*?})", output)
|
||
|
|
for m in matches:
|
||
|
|
comps = m[1:-1].split(",")
|
||
|
|
test = dict(
|
||
|
|
comment = comps[0].strip()[1:-1],
|
||
|
|
input = get_string(comps[1]),
|
||
|
|
output = get_string(comps[2])
|
||
|
|
)
|
||
|
|
if len(comps) >= 4:
|
||
|
|
test["profile"] = get_string(comps[3])
|
||
|
|
if len(comps) >= 5:
|
||
|
|
test["flags"] = comps[4].strip()
|
||
|
|
if len(comps) >= 6:
|
||
|
|
test["rc"] = comps[5].strip()
|
||
|
|
Tests.append(test)
|
||
|
|
|
||
|
|
print json.dumps(Tests)
|