python - Python: これら 2 つの関数を互いに区別する方法

Question

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

def test1():
    return a()

def test2(x):
    return x

Pythonインタープリターでこれを行いたい：

test1()

test2(a())

したがって、これら 2 つの関数 (test1()、test2(x)) が異なる結果を返すように a() を記述する必要があります。

たとえば、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"