人人都爱Python,自动化测试-unittest框架学习五【Gtalent】

#第94行
import StringIO 修改为：import io

#第539行
self.outputBuffer = StringIO.StringIO() 修改为：self.outputBuffer = io.StringIO()

#第631行
print >>sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime)修改为：print(sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime))

#第642行
if not rmap.has_key(cls)：修改为：if not cls in rmap:

#第766行
uo = o.decode('latin-1')修改为：uo = o

#第772行
ue = e.decode('latin-1')修改为：ue = e

#使用TextTestRuner运行
    unittest.TextTestRunner(suite)

# 使用HTMLTestRunner运行
    # 创建 html 文件
    name = open(os.getcwd()+"/report,"wb")
    # 创建运行器实例
    runner = HTMLTestRunner(stream=name,                       # 输出到文件
                            title="Python自动化测试报告",     # html title
                            description="报告详情如下:")      #  描述
    runner.run(suite)
    # unittest.main(testRunner=runner,testcases=suite)
​

# my_math.py

def fib(n = 0)->int:
    """ 斐波那契数列 """
    a, b = 0, 1
    for i in range(n):
        a, b = b, a+b
    return a

def NarcissisticNumber(left=100, right=999)->list:
    """ 水仙花数 """
    assert(left >= 0 and right <= 999)
    lst = []
    for i in range(left, right+1):
        s = str(i)
        one = int(s[-1])
        ten = int(s[-2])
        hun = int(s[-3])
        if i == one ** 3 + ten ** 3 + hun ** 3:
            lst.append(i)
    return lst

if __name__ == '__main__':
    print("斐波那契数列第10项是：",fib(10))

    print("10 - 1000 的水仙花数有：\n",NarcissisticNumber(100,999))

import unittest

from 测试.测试实例 import my_math


class MyMathTest(unittest.TestCase):

    def test_fib(self):
        print("测试",my_math.fib.__doc__)
        a, b = 0, 1
        for i in range(0, 50):
            #  输出结果
            # print("斐波那契数列第{0}项是{1}".format(i, my_math.fib(i)))
            self.assertEqual(my_math.fib(i), a, "断言出错..")
            a , b = b , a+b

    def test_NarcissisticNumber(self):
        print("测试",my_math.NarcissisticNumber.__doc__)
        # 三位水仙花数
        lst = [153,370,371,407]
        res = my_math.NarcissisticNumber()
        self.assertEqual(lst, res, "断言出错..")

if __name__ == '__main__':
    unittest.main(verbosity=2)