def assert_type(code, expected_type):
pb = InteractivePB()
for statement in parse_blob(code):
pb.define(statement)
if statement["name"]:
t = str(pb.pipes[statement["name"]].type)
else:
t = str(pb.pipes[""][0].type.output)
assert t == expected_type, "Expected type %s, got type %s" % (expected_type, t)
def _make_pb(code):
pb = InteractivePB()
for statement in parse_blob(code):
pb.define(statement)
return pb
def exec_blob(blob):
pb = Plumbum()
for st in parse_blob(blob):
pb.define(st)
pb.run()
