的Python：如何區別這兩種功能相互

Question

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號。

Answer 1

import inspect 

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