使用其他函数的输入（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()