Transformations Review

from collections.abc import Callable


def multiply(x: int, y: int) -> int:
    return x * y


def add(x: int, y: int) -> int:
    return x + y


def self_math(math_func: Callable[[int, int], int]) -> Callable[[int], int]:
    # inner_func is defined inside self_math.
    # It can only be referenced directly
    # inside self_math's scope. However, it is then
    # returned and can be captured into a new variable
    # like square_func or double_func, and called that way
    def inner_func(x: int) -> int:
        return math_func(x, x)

    return inner_func


square_func: Callable[[int], int] = self_math(multiply)
double_func: Callable[[int], int] = self_math(add)

print(square_func(5))
# 25

print(double_func(5))
# 10