初学者 Python 帮助 - 主函数

Question

我在处理这段代码时遇到了一些问题：

def sumList(nums):
    sum = 0
    for num in nums:
        sum = sum + num
    return sum
print(sumList([5, 2, 4, 7])
    def numDict():
    num = dict()
    num = 5
    num = 2
    num = 4
    num = 7
print(sumList([5, 2, 4, 7]))
def main():
    nums = sumList()
    print(num[5])
    print(num[2])
    print(num[4])
    print(num[7])
main()

我正在尝试做的是使用 main 函数多次测试 sumList。上面，这些数字加起来产生了 18。我想合并主功能。我该怎么做？

Answer 1

这样的事情怎么样？

def sumList(nums):
    sum = 0
    for num in nums:
        sum = sum + num
    return sum
def main():
    print("Test One:")
    print(sumList([5,2,4,7]))
    print("Test Two:")
    print(sumList([1,2,3,4]))
    print("Test Three:")
    print(sumList([0.5, 0.5, 0.5, 0.5]))

这应该产生这样的输出

Test One:
18
Test Two:
10
Test Three:
2.0

Answer 2

好的，

所以我不是 100% 确定你在说什么，但如果你只是说我如何使用 main 函数测试 sumList，那么你可以做如下的事情：

testCases = [([1,2,3],6) , ([1,1,1],3), ([10,10,12], 32)]
def main():
    for testCase, answer in testCases:
        if sumList(testCase) != answer:
            print("False")
    print("Everything checks out")
# then to actually call your main function like something similar to cpp
if __name__ == '__main__':
    main()

if __name__ == '__main__':约定通常是如何在 python 中访问您的主调用。

如果您对主要功能的重点感到好奇，这里有一个非常好的 Stack 链接。再次堆叠救援。大量使用它！