node registration

node_types.py

import sys
from joint import Joint
from transform import Transform
from unicode_delegate import UnicodeDelegate
from dag_node import DagNode
from curve import Curve

REGISTERED_NODES = {}


def register_node(node_class):
    #setattr(sys.modules[__name__], node_class.__name__, node_class)
    REGISTERED_NODES[node_class.__name__] = node_class
    return node_class

test_node_types.py

import unittest

class TestNodeTypes(unittest.TestCase):

    def test_imported_nodes(self):
      import node_types
      for registered_node_name in REGISTERED_NODES.keys():
        self.assertTrue(hasattr(node_types, registered_node_name))

if __name__ == '__main__':
    unittest.main()