Использовать ввод из другой функции (python 3)

Question

Я действительно новичок в Python, и английский не мой родной язык, поэтому, пожалуйста, будьте добры ко мне.

Как я могу использовать ввод из другой функции?Проблема в том, что я должен использовать TDD, и мои тесты для malt_choice пройдены, поэтому я хотел бы сохранить эту функцию, если это возможно.Спасибо!

def add_malt_menu():
    print('=' * 24)
    print("""
1. CaraRuby-Malt
2. Crystal-Malt
3. PaleAle-Malt
4. Pilsner-Malt
5. Return to main menu""")
   print('=' * 24)

>>>I would like to use the input from malt_choice here<<<

         pseudocode
         if input from maltchoice ==1:
             blablbla
         elif: input from maltchoice ==2:
         blablabla

def malt_choice():
    return input()