def test_get_node_by_name(self): name = "test_name" self.assertEqual(_create_test_node(name=name), Nodzgraph.get_node_by_name(name)) name = "another_test_name" self.assertEqual(_create_test_node(name=name), Nodzgraph.get_node_by_name(name))
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"])