def test__one_layer_tree_w_meta(): smpl_1.uuid = 'one' smpl_2.uuid = 'two' smpl_3.uuid = 'three' assert(smpl_1.uuid == 'one') treefunc = one_layer_tree_w_meta(smpl_1, *[smpl_2, smpl_3]) assert(treefunc.funcmeta == {'smpl_1': ('smpl_2', 'smpl_3')})
def test__one_layer_tree_w_meta__two(): def foo(stuff): return stuff tree_func = one_layer_tree_w_meta(foo, smpl_1, smpl_2, smpl_3) res = tree_func('monkey') assert(res['smpl_1'] == tuple(['monkey_one'])) assert(res['smpl_2'] == tuple(['monkey_two'])) assert(res['smpl_3'] == tuple(['monkey_three']))
def test__one_layer_tree_w_meta__three(): def foo(stuff): return stuff tree_func = one_layer_tree_w_meta(foo, smpl_1, smpl_2, smpl_3) assert(tree_func.funcmeta == {'foo': ('smpl_1', 'smpl_2', 'smpl_3')})