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 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_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 __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_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_()