用python输入两个不同的测试用例

Question

我在使用python测试不同的用例时遇到了问题

• 测试用例1：我必须在不同的行中使用两个字符串
• 测试用例2：我必须在同一行中输入两个字符串

如果我将代码写入第一个测试用例，则第二个测试用例未执行；如果我将代码写入第二个测试用例，则第一个测试用例未执行。

我在c / c ++ 14中试过，两个测试用例都完美无缺，但是在python 3.7中它显示了错误

这是针对测试用例1的

s1=input()
s2=input()

这是针对测试用例2

s1,s2=input().split()

如何为两个测试用例编写代码以获得满意的结果

Answer 1

您可能会因尝试解包过多的元素而捕获错误，并要求其他输入：

def get_inputs():
    s1 = input()
    try:
        s1, s2 = s1.split()
    except ValueError:
        s2 = input()
    return s1, s2