def test_create_connections(self): node_setup = _create_nodes_setup() Nodzgraph._create_attributes(self._test_attrs_data) Nodzgraph._create_connections(self._test_cons_data) expected_connections = [ (x, y) for x, y in self._test_cons_data.iteritems() if (x.split(".")[0] in node_setup and y.split(".")[0] in node_setup ) ] self.assertListEqual(sorted(expected_connections), sorted(Nodzgraph.graph.evaluateGraph()))
def test_create_attributes(self): node_setup = _create_nodes_setup() x = Nodzgraph.all_node_names # create the attributes Nodzgraph._create_attributes(self._test_attrs_data) # compare created attributes with the expected from _test_attrs_data # only considering nodes in node_to_create for attr, attr_data in self._test_attrs_data.iteritems(): node_name = attr.split(".")[0] attr_name = attr.split(".")[1] if node_name in node_setup: self.assertIn(node_name, Nodzgraph.nodes_dict) node = Nodzgraph.get_node_by_name(node_name) if attr_data["type"] == ("plug" or "slot"): self.assertIn(attr_name, node.plugs) self.assertEqual(node.plugs[attr_name].dataType, attr_data["data_type"]) if attr_data["type"] == ("socket" or "slot"): self.assertIn(attr_name, node.sockets) self.assertEqual(node.sockets[attr_name].dataType, attr_data["data_type"])