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)