def print_tree(tree: SExp) -> str:
a = tree.as_atom()
if a is not None:
if len(a) == 0:
return "() "
return "%d " % a[0]
ret = "("
for i in tree.as_pair():
ret += print_tree(i)
ret += ")"
return ret
def test_arbitrary_underlying_tree(self):
# SExp provides a view on top of a tree of arbitrary types, as long as
# those types implement the CLVMObject protocol. This is an example of
# a tree that's generated
class GeneratedTree:
depth: int = 4
val: int = 0
def __init__(self, depth, val):
assert depth >= 0
self.depth = depth
self.val = val
@property
def atom(self) -> Optional[bytes]:
if self.depth > 0:
return None
return bytes([self.val])
@property
def pair(self) -> Optional[Tuple[Any, Any]]:
if self.depth == 0:
return None
new_depth: int = self.depth - 1
return (GeneratedTree(new_depth, self.val),
GeneratedTree(new_depth, self.val + 2**new_depth))
tree = SExp.to(GeneratedTree(5, 0))
assert print_leaves(tree) == "0 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 "
tree = SExp.to(GeneratedTree(3, 0))
assert print_leaves(tree) == "0 1 2 3 4 5 6 7 "
tree = SExp.to(GeneratedTree(3, 10))
assert print_leaves(tree) == "10 11 12 13 14 15 16 17 "
def test_wrap_sexp(self):
# it's a bit of a layer violation that CLVMObject unwraps SExp, but we
# rely on that in a fair number of places for now. We should probably
# work towards phasing that out
o = CLVMObject(SExp.to(1))
assert o.atom == bytes([1])
def test_cast_1(self):
# this was a problem in `clvm_tools` and is included
# to prevent regressions
sexp = SExp.to(b"foo")
t1 = sexp.to([1, sexp])
validate_sexp(t1)
def test_eager_conversion(self):
with self.assertRaises(ValueError):
SExp.to(("foobar", (1, {})))
def test_empty_list_conversions(self):
a = SExp.to([])
assert a.as_atom() == b""
def test_none_conversions(self):
a = SExp.to(None)
assert a.as_atom() == b""
def test_int_conversions(self):
a = SExp.to(1337)
assert a.as_atom() == bytes([0x5, 0x39])
def test_string_conversions(self):
a = SExp.to("foobar")
assert a.as_atom() == "foobar".encode()
def test_list_conversions(self):
a = SExp.to([1, 2, 3])
assert print_tree(a) == "(1 (2 (3 () )))"