def test_create_register_node(self):
node = control.create_register_node(name='temp')
self.assertTrue(isinstance(node, pm.nt.Transform))
self.assertEqual(node.name(), 'temp_reg_node')
for attr in ['reg_node', 'version']:
self.assertTrue(hasattr(node, attr))
self.assertEqual(node.getParent(), None)
def test_register_object(self):
obj = pm.joint(n='jnt_1')
node = control.create_register_node(name='temp')
control.register_object(node, 'temp', obj)
self.assertEqual(node.temp.listConnections()[0], obj)
