函数测试

这里以体育竞技模拟程序内涵数为例

原完整代码

# -*- coding: utf-8 -*-
"""
Created on Wed May 15 11:38:02 2019

@author: lenovo
"""

import random
import math
def printIntro():
    print("这个程序模拟量个选手A和B的乒乓球比赛")
    print("程序运行需要A和B的能力值（以0到1之间的小数表示）")
    print("作者：呆。   （02）")
def getInputs():
    a = eval(input("请输入选手A的能力值（0-1）: "))
    b = eval(input("请输入选手B的能力值（0-1）: "))
    n = eval(input("模拟比赛的场次: "))
    return a, b, n
 
def printSummary(winsA, winsB):
    n = winsA + winsB
    print("竞技分析开始， 共模拟{}场比赛".format(n))
    print("选手A获胜{}场比赛， 占比{:0.1%}".format(winsA, winsA/n))
    print("选手B获胜{}场比赛， 占比{:0.1%}".format(winsB, winsB/n))
 
def gameOver(a, b):
    return (a==11 and b<10) or (b==11 and a<10) or (a>=10 and b>=10 and math.fabs(a-b)==2)
    
def simoneGame(probA, probB):
    scoreA, scoreB = 0, 0
    if random.random() < 0.5:
        serving = "A"
    else :
        serving = "B"
    while not gameOver(scoreA, scoreB):
        if serving == "A":
            if random.random() < probA:
                scoreA += 1
            else:
                serving = "B"
        else:
            if random.random() < probB:
                scoreB += 1
            else:
                serving = "A"
    return scoreA, scoreB
def simOneGame(probA, probB):
    winsA, winsB = 0, 0
    for i in range(7):
        scoreA, scoreB = simoneGame(probA, probB)
        if scoreA > scoreB:
            winsA += 1
        else:
            winsB += 1
    return winsA, winsB
def simNGames(n ,probA, probB):
    winsA, winsB = 0, 0
    for i in range(n):
        scoreA, scoreB = simOneGame(probA, probB)
        if scoreA > scoreB:
            winsA += 1
        else:
            winsB += 1
    return winsA, winsB
 
def main():
    printIntro()
    probA, probB, n = getInputs()
    winsA, winsB = simNGames(n, probA, probB)
    printSummary(winsA, winsB)
main()

分离函数单独测试

首先测试最底层函数 gameOver(a, b)

import math
def gameOver(a, b):
    return (a==11 and b<10) or (b==11 and a<10) or (a>=10 and b>=10 and math.fabs(a-b)==2)
print(gameOver(9,11))
print(gameOver(11,7))
print(gameOver(13,11))
print(gameOver(11,11))
print(gameOver(10,11))

结果正常：

 由于gameOver函数无误，故借用该函数测试函数simoneGame(probA, probB)

import random
import math
def gameOver(a, b):
    return (a==11 and b<10) or (b==11 and a<10) or (a>=10 and b>=10 and math.fabs(a-b)==2)
    
def simoneGame(probA, probB):
    scoreA, scoreB = 0, 0
    if random.random() < 0.5:
        serving = "A"
    else :
        serving = "B"
    while not gameOver(scoreA, scoreB):
        if serving == "A":
            if random.random() < probA:
                scoreA += 1
            else:
                serving = "B"
        else:
            if random.random() < probB:
                scoreB += 1
            else:
                serving = "A"
    return scoreA, scoreB
a,b=simoneGame(0.5, 0.5)
print(a)
print(b)

多次运行结果：

与期望相同

在借用上述函数测试函数simOneGame(probA, probB)

import random
import math
def gameOver(a, b):
    return (a==11 and b<10) or (b==11 and a<10) or (a>=10 and b>=10 and math.fabs(a-b)==2)
    
def simoneGame(probA, probB):
    scoreA, scoreB = 0, 0
    if random.random() < 0.5:
        serving = "A"
    else :
        serving = "B"
    while not gameOver(scoreA, scoreB):
        if serving == "A":
            if random.random() < probA:
                scoreA += 1
            else:
                serving = "B"
        else:
            if random.random() < probB:
                scoreB += 1
            else:
                serving = "A"
    return scoreA, scoreB
def simOneGame(probA, probB):
    winsA, winsB = 0, 0
    for i in range(7):
        scoreA, scoreB = simoneGame(probA, probB)
        if scoreA > scoreB:
            winsA += 1
        else:
            winsB += 1
    return winsA, winsB
a,b=simOneGame(0.5, 0.5)
print(a)
print(b)

结果如下：

竞技规则为7局4胜

故结果出错改动如下：

def simOneGame(probA, probB):
    winsA, winsB = 0, 0
    for i in range(7):
        scoreA, scoreB = simoneGame(probA, probB)
        if winsA==4 or winsB==4:
            continue
        elif scoreA > scoreB:
            winsA += 1
        else:
            winsB += 1
    return winsA, winsB

改动后结果正确

同理嵌套检测函数simNGames(n ,probA, probB)及main函数

过程:略