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):
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 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 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.output = plugs.OutputPlug("output", self)
self.valuePlug_ = plugs.InputPlug("value", self, value=[])
self.addPlug(self.output, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.plugAffects(self.valuePlug_, self.output)
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.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=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.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 testAttributeAffectsDirty(self):
newNode = sum.SumNode("addNode")
valuePlug = newNode.getPlug("value1")
newPlug = plug.OutputPlug(name="testPlug", node=newNode)
newNode.addPlug(newPlug, clean=True)
newNode.plugAffects(newNode.getPlug("value2"), newPlug)
self.assertEquals(newNode.getPlug("value1").dirty, False)
valuePlug.value = 10
self.assertEquals(newPlug.dirty, False)
newNode.compute(newNode.getPlug("output"))
self.assertEquals(newPlug.dirty, False)
newNode.getPlug("value2").value = 50
self.assertEquals(newPlug.dirty, True)
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_ = 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.valuePlug_ = plugs.InputPlug("value", self, value={})
self.valueToAddPlug_ = plugs.InputPlug("valueToAdd", self, value={})
self.keyPlug_ = plugs.InputPlug("key", self, value={})
self.addPlug(self.outputPlug_, clean=True)
self.addPlug(self.valuePlug_, clean=True)
self.addPlug(self.valueToAddPlug_, clean=True)
self.addPlug(self.keyPlug_, clean=True)
self.plugAffects(self.valuePlug_, self.outputPlug_)
self.plugAffects(self.valueToAddPlug_, self.outputPlug_)
self.plugAffects(self.keyPlug_, 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 testDisconnect(self):
floatTypeAttr = plugs.OutputPlug(name="testOutput")
self.plug.connect(floatTypeAttr)
self.plug.disconnect(floatTypeAttr)
self.assertEquals(len(self.plug), 0)
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)
def initialize(self):
baseNode.BaseNode.initialize(self)
self.outputPlug_ = plugs.OutputPlug("output", self)
self.addPlug(self.outputPlug_, clean=True)
