我怎样才能使选项B正常工作，因为最后一个答案并没有完全起作用

Question

我需要它，所以如果你选择选项B，你可以输入一个数字，它会告诉你它是否是浮点数。有点像如果你选择选项A并输入一个数字，它会让你输入一个数值并告诉你它是否是整数。

这是我的代码：

def getNumber():
while True:
try:
userInput = int(input("What number would like to test as an integer ? "))
return userInput
except ValueError as ve:
print("Not an integer! Try again.")
continue
print("What would you like to do?")
print("Option A = Interger")
print("Option B = Floating Point")
Msg = input()
if Msg == 'A':
integer_received = getNumber()
print(integer_received, "is an int !")
else:
print("You did not choose 'A' or 'B', make sure it is capitalized!")

Answer 1

def getNumber():
while True:
try:
userInput = int(input("What number would like to test as an integer ? "))
return userInput
except ValueError as ve:
print("Not an integer! Try again.")
continue
def getFloatintNumber():
while True:
try:
userInput = float(input("What number would like to test as float ? "))
return userInput
except ValueError as ve:
print("Not a float! Try again.")
continue

print("What would you like to do?")
print("Option A = Interger")
print("Option B = Floating Point")
Msg = input()
if Msg == 'A':
integer_received = getNumber()
print(integer_received, "is an int !")
elif Msg == 'B':
float_received = getFloatintNumber()
print(float_received, "is a float !")
else:
print("You did not choose 'A' or 'B', make sure it is capitalized!")