def test_create_curve_new_reg_node(self):
reg_node, cnt = control.create_curve(name='temp',
typ='circle',
scale=3.0,
color='light_blue')
for s in cnt.getShapes():
self.assertEqual(s.overrideColor.get(), 6)
self.assertEqual(reg_node.temp.listConnections()[0], cnt)
self.assertTrue(isinstance(cnt, pm.nt.Transform))
self.assertTrue(isinstance(cnt.getShape(), pm.nt.NurbsCurve))
def test_create_curve_existing_reg_node(self):
reg_node = pm.group(empty=1)
reg_node.addAttr('version', dt="string")
reg_node.addAttr('reg_node', dt="string")
reg_node2, cnt = control.create_curve(name='temp',
typ='circle',
scale=3.0,
color='light_blue',
reg_node=reg_node)
for s in cnt.getShapes():
self.assertEqual(s.overrideColor.get(), 6)
self.assertEqual(reg_node, reg_node2)
self.assertEqual(reg_node.temp.listConnections()[0], cnt)
self.assertTrue(isinstance(cnt, pm.nt.Transform))
self.assertTrue(isinstance(cnt.getShape(), pm.nt.NurbsCurve))
