def test_match_object(self):
a = pm.circle()[0]
b = pm.circle()[0]
b.rx.set(75)
b.ry.set(75)
b.rz.set(75)
self.assertFalse(transforms.assertOrientationMatches(a, b))
control.match_object(a, b)
self.assertTrue(transforms.assertOrientationMatches(a, b))
def test_make_control(self):
controls = []
reg_node = None
for i in range(5):
if i == 0:
reg_node, control = fk_rig.make_control(name='temp%s' %
str(i + 1),
obj=self.jnts[i],
buffers=2,
color='light_blue',
scale=3.0,
typ='cube')
else:
reg_node, control = fk_rig.make_control(name='temp%s' %
str(i + 1),
obj=self.jnts[i],
buffers=2,
color='light_blue',
scale=3.0,
typ='cube',
reg_node=reg_node)
controls.append(control)
self.assertEqual(len(controls), 5)
count = 0
for cnt, jnt in zip(controls, self.jnts):
name = 'temp%s' % (count + 1)
for s in cnt.getShapes():
self.assertEqual(s.overrideColor.get(), 6)
self.assertTrue(transforms.assertOrientationMatches(cnt, jnt))
attr = getattr(reg_node, cnt.name())
self.assertEqual(attr.listConnections()[0], cnt)
self.assertEqual(cnt.name().split('_')[0], name)
self.assertTrue(transforms.assertLocationsMatch(cnt, jnt))
self.assertTrue(transforms.assertAllZero(cnt))
# Check that joint is under curve
pm.select(cnt, hi=True, r=True)
sel = pm.ls(sl=1)
self.assertIn(jnt, sel)
pm.select(clear=True)
# Check heirarchy
try:
p1 = cnt.getParent()
p2 = p1.getParent()
if count > 0:
prnt = p2.getParent()
self.assertEqual(prnt, self.jnts[count - 1])
self.assertTrue(transforms.assertAllZero(p1))
self.assertEqual(p1.name(), '%s_btm_node' % name)
self.assertEqual(p2.name(), '%s_top_node' % name)
except Exception as e:
str_1 = 'Heirarchy check failed: Error: ', str(e)
self.fail(str_1)
count += 1
def test_make_control(self):
controls = []
reg_node = None
for i in range(5):
if i == 0:
reg_node, control = fk_rig.make_control(name='temp%s' %
str(i+1),
obj=self.jnts[i],
buffers=2,
color='light_blue',
scale=3.0,
typ='cube')
else:
reg_node, control = fk_rig.make_control(name='temp%s' %
str(i+1),
obj=self.jnts[i],
buffers=2,
color='light_blue',
scale=3.0,
typ='cube',
reg_node=reg_node)
controls.append(control)
self.assertEqual(len(controls), 5)
count = 0
for cnt, jnt in zip(controls, self.jnts):
name = 'temp%s' % (count+1)
for s in cnt.getShapes():
self.assertEqual(s.overrideColor.get(), 6)
self.assertTrue(transforms.assertOrientationMatches(cnt, jnt))
attr = getattr(reg_node, cnt.name())
self.assertEqual(attr.listConnections()[0], cnt)
self.assertEqual(cnt.name().split('_')[0], name)
self.assertTrue(transforms.assertLocationsMatch(cnt, jnt))
self.assertTrue(transforms.assertAllZero(cnt))
# Check that joint is under curve
pm.select(cnt, hi=True, r=True)
sel = pm.ls(sl=1)
self.assertIn(jnt, sel)
pm.select(clear=True)
# Check heirarchy
try:
p1 = cnt.getParent()
p2 = p1.getParent()
if count > 0:
prnt = p2.getParent()
self.assertEqual(prnt, self.jnts[count-1])
self.assertTrue(transforms.assertAllZero(p1))
self.assertEqual(p1.name(), '%s_btm_node' % name)
self.assertEqual(p2.name(), '%s_top_node' % name)
except Exception as e:
str_1 = 'Heirarchy check failed: Error: ', str(e)
self.fail(str_1)
count += 1
def test_assertOrientationMatches(self):
a = pm.circle()[0]
b = pm.circle()[0]
self.assertTrue(transforms.assertOrientationMatches(a, b))
b.rx.set(90)
self.assertFalse(transforms.assertOrientationMatches(a, b))
def test_assertOrientationMatches(self):
a = pm.circle()[0]
b = pm.circle()[0]
self.assertTrue(transforms.assertOrientationMatches(a, b))
b.rx.set(90)
self.assertFalse(transforms.assertOrientationMatches(a, b))
