def test_schema_connections():
schema = Schema()
n1 = Node()
n2 = Node()
n3 = Node()
schema.add_node(n1)
schema.add_node(n2)
assert_raises(ValueError, schema.connect_nodes, n1, n1)
schema.connect_nodes(n1, n2)
assert((n1, n2) in schema.connections)
assert(n1.out_conn[0] is n2)
assert(n2.in_conn[0] is n1)
assert_raises(ValueError, schema.disconnect_nodes, n2, n1)
schema.connect_nodes(n3, n2)
schema.disconnect_nodes(n1, n2)
assert(schema.connections == [(n3, n2)])
assert(n1.out_conn == [])
assert(n2.in_conn == [n3])
def test_schema():
schema = Schema()
n1 = Node()
n2 = Node()
n3 = Node()
schema.add_node(n1)
schema.add_node(n2)
schema.add_node(n3)
assert(n1 in schema.nodes)
assert(n2 in schema.nodes)
schema.delete_node(n1)
assert(n1 not in schema.nodes)
def __init__(self):
super(SumLabelNode, self).__init__()
self.node_type = 'Sum Label'
self.accepts_input = True
self.lbl = QLabel()
def get_view(self):
return WidgetNodeView(self)
def get_widget(self):
return self.lbl
def update(self):
s = sum((n.get() for n in self.in_conn))
self.lbl.setText(str(s))
if __name__ == '__main__':
app = QApplication([])
sch = Schema()
sv = SchemaView(sch)
n = SpinBoxNode()
n2 = SpinBoxNode()
n3 = SumLabelNode()
sch.add_node(n)
sch.add_node(n2)
sch.add_node(n3)
gv = QGraphicsView(sv)
gv.show()
app.exec_()
def test_schema_tofile():
from StringIO import StringIO
s = Schema()
n1 = Node()
n2 = Node()
s.add_node(n1)
s.add_node(n2)
s.connect_nodes(n1, n2)
f = StringIO()
s.to_disk(f)
f.seek(0)
s2 = Schema()
s2.from_disk(f)
assert(len(s.connections) == len(s2.connections))
assert(len(s.nodes) == len(s2.nodes))
def test_schema_exception():
schema = Schema()
n1 = Node()
schema.add_node(n1)
schema.add_node(n1)
class SumLabelNode(Node):
def __init__(self):
super(SumLabelNode, self).__init__()
self.node_type = 'Sum Label'
self.accepts_input = True
self.lbl = QLabel()
def get_view(self):
return WidgetNodeView(self)
def get_widget(self):
return self.lbl
def update(self):
s = sum((n.get() for n in self.in_conn))
self.lbl.setText(str(s))
if __name__ == '__main__':
app = QApplication([])
sch = Schema()
sv = SchemaView(sch)
n = SpinBoxNode()
n2 = SpinBoxNode()
n3 = SumLabelNode()
sch.add_node(n)
sch.add_node(n2)
sch.add_node(n3)
gv = QGraphicsView(sv)
gv.show()
app.exec_()