Как сделать утверждение, если функция была вызвана для вложенной функции?

Question

Я пытаюсь сделать функциональную проверку моего скрипта.Поэтому я хочу проверить, все ли функции были вызваны, не превращая их в класс с какими-то скрытыми переменными или разбивая тест на несколько разделенных.*

iety_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 после вызова sort_functions.