Python根据不同输入值调用不同函数

根据参数不同值去调用不同的函数处理

def fun_a():
    print 'a'
def fun_b():
    print 'b'
def fun_z():
    print 'z'
def test_function(input_key):
    function_map = {
                     'a':fun_a,
                     'b':fun_b,
                     'z':fun_z,
                    }
    return function_map[input_key]()