如果为嵌套函数调用了函数，如何进行断言？

Question

我正在尝试对脚本进行功能测试。因此，我想检查是否调用了所有函数，而没有将它们设为带有一些隐藏变量的类，也没有将测试分为几个单独的部分。

有示例代码显示了我的问题： typ.py

from variety_02 import variety_functions


def variety():
    inp = input("text")
    if inp == "test":
        variety_functions()

variety_02.py

def test_02():
    print("\nTest 02 called!")


def test_01():
    print("\nTest 01 called!")


def variety_functions():
    test_01()
    test_02()

test_variety.py

from variety import variety, variety_functions
from unittest import mock


def test_variety(monkeypatch):
    with mock.patch("variety.variety_functions", wraps=variety_functions) as mock_automatic_execution:
        monkeypatch.setattr("builtins.input", lambda x: "test")
        variety()
        mock_automatic_execution.assert_called()

我仍在寻找一种方法来断言是否在综函数调用之后调用了test_01和test_02函数。