def setUp(self):
self.graph = graph.Graph(name="testGraph")
self.testNode = baseNode.BaseNode("testNode")
self.testNode2 = baseNode.BaseNode("testNode2")
self.testNode.addPlug(plugs.InputPlug("testInput", self.testNode))
self.testNode.addPlug(plugs.OutputPlug("testOutput", self.testNode))
self.testNode2.addPlug(plugs.InputPlug("testInput1", self.testNode2))
self.testNode2.addPlug(plugs.OutputPlug("testOutput1", self.testNode2))
def initialize(self):
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=[])
self.searchValuePlug_ = plugs.InputPlug("searchValue", self, value="")
self.replaceValuePlug_ = plugs.InputPlug("replaceValue", self, value=None)
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.directoriesPlug_ = plugs.InputPlug("directories", self, [])
self.recursivePlug_ = plugs.InputPlug("recursive", self, True)
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.directoriesPlug_, clean=True)
self.addPlug(self.recursivePlug_, clean=True)
self.plugAffects(self.directoriesPlug_, self.outputPlug_)
self.plugAffects(self.recursivePlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.directoriesPlug_ = plugs.InputPlug("directories", self)
self.fullpathPlug_ = plugs.InputPlug("fullPath", self)
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.directoriesPlug_, clean=True)
self.addPlug(self.fullpathPlug_, clean=True)
self.plugAffects(self.directoriesPlug_, self.outputPlug_)
self.plugAffects(self.fullpathPlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.addPlug(self.outputPlug_, clean=True)
self.value1Plug_ = plugs.InputPlug("value1", self, value=0)
self.value2Plug_ = plugs.InputPlug("value2", self, value=0)
self.addPlug(self.value1Plug_, clean=True)
self.addPlug(self.value2Plug_, clean=True)
self.plugAffects(self.value1Plug_, self.outputPlug_)
self.plugAffects(self.value2Plug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=[])
self.searchPlug_ = plugs.InputPlug("searchValue", self, value="")
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.addPlug(self.searchPlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
self.plugAffects(self.searchPlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=0)
self.moduloBy = plugs.InputPlug("moduloBy", self, value=0)
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.addPlug(self.moduloBy, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
self.plugAffects(self.moduloBy, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.sourcePlug_ = plugs.InputPlug("source", self, [])
self.destinationPlug_ = plugs.InputPlug("destination", self, [])
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.sourcePlug_, clean=True)
self.addPlug(self.destinationPlug_, clean=True)
self.plugAffects(self.sourcePlug_, self.outputPlug_)
self.plugAffects(self.destinationPlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value={})
self.keyPlug_ = plugs.InputPlug("key", self, value="")
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.addPlug(self.keyPlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
self.plugAffects(self.keyPlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=[])
self.delimiterPlug_ = plugs.InputPlug("delimiter", self, value="")
self.returnIndexPlug_ = plugs.InputPlug("returnIndex",
self,
value=None)
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.sourceFile_ = plugs.InputPlug("sourceFiles", self, value=[])
self.destinationDirectoryPlug_ = plugs.InputPlug(
"destinationDirectory", self, value="")
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.sourceFile_, clean=True)
self.addPlug(self.destinationDirectoryPlug_, clean=True)
self.plugAffects(self.sourceFile_, self.outputPlug_)
self.plugAffects(self.destinationDirectoryPlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = self.addPlug(self.outputPlug_, clean=True)
self.conditionPlug_ = plugs.InputPlug("condition", self, value=True)
self.ifTruePlug_ = plugs.InputPlug("ifTrue", self, value=0)
self.ifFalsePlug = plugs.InputPlug("ifFalse", self, value=0)
self.addPlug(self.conditionPlug_, clean=True)
self.addPlug(self.ifTruePlug_, clean=True)
self.addPlug(self.ifFalsePlug, clean=True)
self.plugAffects(self.conditionPlug_, self.outputPlug_)
self.plugAffects(self.ifTruePlug_, self.outputPlug_)
self.plugAffects(self.ifFalsePlug, self.outputPlug_)
def testSerializeNodePlugsToGraph(self):
node = baseNode.BaseNode("testNode")
node2 = baseNode.BaseNode("testNode2")
node.addPlug(plugs.InputPlug("testInputPlug", node=node))
node.addPlug(plugs.OutputPlug("testOutputPlug", node=node))
node2.addPlug(plugs.InputPlug("testInputPlug", node=node2))
node2.addPlug(plugs.OutputPlug("testOutputPlug", node=node2))
node2.getPlug("testOutputPlug").connect(node.getPlug("testInputPlug"))
self.graph.addNode(node)
self.graph.addNode(node2)
savedGraph = self.graph.serializeGraph()
self.newGraph = graph.Graph.loadGraph(savedGraph)
self.assertEquals(len(self.graph), len(self.newGraph))
self.assertEquals(self.graph._name, self.newGraph._name)
self.assertDictEqual(savedGraph, self.newGraph.serializeGraph())
def initialize(self):
self.outputPlug_ = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=float(0))
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.addPlug(self.outputPlug_, clean=True)
self.arrayPlug_ = plugs.OutputPlug("array", self, value=[])
self.indexPlug_ = plugs.InputPlug("index", self, value=0)
self.addPlug(self.arrayPlug_, clean=True)
self.addPlug(self.indexPlug_, clean=True)
self.plugAffects(self.arrayPlug_, self.outputPlug_)
self.plugAffects(self.indexPlug_, self.outputPlug_)
def testConnect(self):
inputAttr = plugs.InputPlug(name="testInput")
outputAttr = plugs.OutputPlug(name="testOutput")
self.plug.connect(outputAttr)
self.assertEquals(self.plug.connections[0].name,
self.plug.name + "_" + outputAttr.name)
self.assertEquals(self.plug.connections[0].input, self.plug)
self.assertEquals(self.plug.connections[0].output, outputAttr)
self.assertTrue(self.plug.isConnected())
self.assertTrue(outputAttr.isConnected())
self.assertTrue(outputAttr.isConnectedTo(self.plug))
self.assertEquals(outputAttr.getConnection(self.plug),
self.plug.connections[0])
# failed to connect as plug already exists
self.assertIsNone(self.plug.connect(inputAttr))
def testGetAttributeAffects(self):
outputPlug1 = plug.OutputPlug("testOutPlug1", self.node)
outputPlug2 = plug.OutputPlug("testOutPlug2", self.node)
inputPlug2 = plug.InputPlug("testinPlug2", self.node)
self.node.plugAffects(self.plug, outputPlug1)
self.node.plugAffects(inputPlug2, outputPlug1)
self.node.plugAffects(inputPlug2, outputPlug2)
self.assertEqual(self.node.getPlugAffects(outputPlug1),
set([self.plug, inputPlug2]))
self.assertEqual(self.node.getPlugAffects(outputPlug2),
set([inputPlug2]))
self.assertEqual(self.node.getPlugAffects(self.plug),
set([outputPlug1]))
self.assertEqual(self.node.getPlugAffects(inputPlug2),
set([outputPlug1, outputPlug2]))
def setUp(self):
self.node = baseNode.BaseNode("testNode")
self.plug = plug.InputPlug("testPlug", self.node)
self.node.addPlug(self.plug)
def testAddPlug(self):
newplug = plug.InputPlug("testPlug1", self.node)
self.node.addPlug(newplug)
self.assertEquals(len(self.node.plugs), 2)
self.node.deletePlug(newplug)
self.assertEquals(len(self.node.plugs), 1)
def setUp(self):
self.plug = plugs.InputPlug(name="testName")
def setUp(self):
self.inputPlug = plug.InputPlug("inputPlug1")
self.outputPlug = plug.OutputPlug("outputPlug1")
self.edge = baseEdge.Edge("testEdge1", inputPlug=self.inputPlug, outputPlug=self.outputPlug)
self.edge2 = baseEdge.Edge("testEdg2", inputPlug=self.inputPlug, outputPlug=self.outputPlug)
