def setUp(self):
'''Create New Scene with group of hierachy spheres'''
setup()
new_obs = []
collumn = 5
row = 7
for i in range(collumn):
new_ob = pm.polySphere(name='testSphere_%s'%i)[0]
if len(new_obs):
#last_ob = new_obs.pop(-1)
new_ob.setParent(new_obs[-1])
new_obs.append(new_ob)
new_obs = [new_obs[0]]
for i in range(row):
dup_ob = pm.duplicate(
new_obs[0],
name = new_obs[0].name()+'_%s'%(i+1),
rc=True)[0]
new_obs.append(dup_ob)
self.genObRoots = new_obs
self.selected = pm.ls('testSphere_*', type='transform')
self.suffix = '_suffix'
self.test_func = {}
for m in [
'single',
'set',
'hierachy',
'oneToOne',
'last',
'singleLast',
'lastType',
'multiType']:
self.test_func[m] = partial(ul.do_function_on,mode=m)
def test_iter_hierachy(self):
joints = []
for i in range(5):
joints.append(pm.joint(p=[0, i, 0]))
results = ul.iter_hierachy(joints[0])
self.assertEqual(
pm.ls(type='joint'), list(results),
'results does not match scene.\nScene:\n%s\nResult:\n%s'%(joints, results))
def run_test(use_wrapper=False):
'''test function with wrapper'''
pm.select(self.genObRoots)
test_wrapper = partial(
test_func, sl=use_wrapper)
if use_wrapper:
test_results = test_wrapper()
else:
test_results = test_wrapper(self.genObRoots[:-1], self.genObRoots[-1])
self.general_test(pm.ls(type='parentConstraint'), test_results)
return test_results
def run_test(use_wrapper=False):
joints = []
locs = []
count = len(self.genObRoots)
for i in range(count):
pm.select(cl=True)
joints.append(pm.joint(p=[0, i, 0]))
loc = pm.spaceLocator()
locs.append(loc)
sph = pm.polySphere(radius=10)
pm.select(cl=True)
for i in range(count):
i += 1
pm.select(sph[0].getShape().vtx[i], add=True)
pm.select(locs, add=True)
pm.select(joints, add=True)
test_wrapper = partial(
test_func, sl=use_wrapper)
if use_wrapper:
test_results = test_wrapper()
else:
test_results = test_wrapper(pm.selected())
self.assertTrue(
pm.ls(type='parentConstraint'),
'Joint and Loc did not connect, No Parent Constraint in Scene')
self.assertTrue(
pm.ls(type='pointOnPolyConstraint'),
'Loc did not connect to mesh, no PointOnPoly Constraint in Scene')
for j, l in zip(joints, locs):
self.assertTrue(
j.inputs(type='parentConstraint'),
'%s do not have any Parent Constraint Connection'%j)
self.assertTrue(
l.outputs(type='parentConstraint'),
'%s do not have any Parent Constraint Connection'%l)
self.assertTrue(
l.inputs(type='pointOnPolyConstraint'),
'%s do not have any Point On Poly Constraint Connection'%l)
return test_results
def run_test(use_wrapper=False):
'''test function with wrapper'''
pm.select(cl=True)
joints = []
for i in range(5):
joints.append(pm.joint(p=[0, i, 0]))
pm.select(self.selected, r=True)
pm.select(joints, add=True)
test_wrapper = partial(
test_func, sl=use_wrapper)
if use_wrapper:
test_results = test_wrapper()
else:
test_results = [
test_wrapper(m,js) for m, js in zip(
[o for o in pm.selected() if ul.get_type(o) == 'mesh'],
[o for o in pm.selected() if ul.get_type(o) == 'joint'])]
self.general_test(pm.ls(type='skinCluster'), test_results)
for skinClter in pm.ls(type='skinCluster'):
self.general_test(skinClter.getInfluence(), joints)
return test_results
def run_test(use_wrapper=False):
'''test function with wrapper'''
pm.select(self.genObRoots)
test_wrapper = partial(
test_func, sl=use_wrapper)
if use_wrapper:
test_results = test_wrapper()
else:
test_results = [
test_wrapper(o, pm.selected()[id+1])
for id, o in enumerate(pm.selected())
if not id%2]
self.general_test(pm.ls(type='parentConstraint'), test_results)
return test_results