如何让我的函数在python中区分输入'I‘和'II’？

def differentiate_input(input_str):
    if len(input_str) == 1:
        # 处理输入为'I'的情况
        # 这里可以编写相应的逻辑代码
        print("输入为'I'")
    elif len(input_str) == 2:
        # 处理输入为'II'的情况
        # 这里可以编写相应的逻辑代码
        print("输入为'II'")
    else:
        # 处理其他情况
        print("输入不符合要求")

# 调用函数进行测试
differentiate_input('I')
differentiate_input('II')