def test_svg_creation_wrapped(self):
opParent = OpParent(graph=Graph())
lazyWrap = OperatorWrapper(
LazyOp,
broadcastingSlotNames=["in_a", "in_b"],
operator_kwargs={"inner_order": inner_order},
parent=opParent,
)
lazyWrap.out_a.resize(2)
lazyWrap.in_a.setValue(self.a)
lazyWrap.in_b.setValue(self.b)
assert lazyWrap.out_a[0][:].wait().shape == outer_shape
assert lazyWrap.out_b[0][:].wait().shape == outer_shape
assert lazyWrap.out_a[1][:].wait().shape == outer_shape
assert lazyWrap.out_b[1][:].wait().shape == outer_shape
generateSvgFileForOperator(os.path.join(self.svg_dir, "lazyWrap.svg"), lazyWrap, 5)
def test_svg_creation_with_parent(self):
opParent = OpParent(graph=Graph())
opParent.setupOutputs()
generateSvgFileForOperator(os.path.join(self.svg_dir, "opParent.svg"), opParent, 5)
def test_svg_creation_simple(self):
lazyOp = LazyOp(graph=Graph(), inner_order=inner_order)
lazyOp.setupOutputs()
generateSvgFileForOperator(os.path.join(self.svg_dir, "lazyOp.svg"), lazyOp, 5)
