Python:如何将这两个函数相互区分开来

时间:2012-06-22 21:52:26

标签: python function

def a():
    """ Need to write something here."""

def test1():
    return a()

def test2(x):
    return x

我想在python解释器中执行此操作:

test1()

test2(a())

所以我需要做的是写一个()来使这两个函数(test1(),test2(x))返回不同的结果。

例如,我希望test1返回数字1,test2返回除数字1之外的其他内容。

1 个答案:

答案 0 :(得分:2)

import inspect

def a():
    s = inspect.stack()
    if s[1][3] == '<module>':
        return "From test2"
    else:
        return "Not from test2"