def test_wireToVertices(self): obj = self.ofactory.getNewObject(4, True) val = self.vfactory.getNewValue() edge = SMGPTEdge(0, val, obj) edge.wireToVertices() ptedges_to_object = obj.getPTEdges() ptedge_from_value = val.getPTEdge() self.assertEqual(len(ptedges_to_object), 1) self.assertIs(ptedges_to_object[0], edge) self.assertIs(ptedge_from_value, edge)
def addPTEdge(self, offset, value, smgObject): edge = SMGPTEdge(offset, value, smgObject) edge.wireToVertices() return edge