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"])
