创建一个名为test_cities.py的程序,对刚编写的函数进行测试(别忘了,你需要导入模块unittest以及要测试的函数)。
编写一个名为test_city_country()的方法,核实使用类似于’santiago’和’chile’这样的值来调用前述函数时,得到的字符串是正确的。
运行test_cities.py,确认测试test_city_country()通过了。
主程序
import unittest
from city_functions import city_functions
class NamesTestCase(unittest.TestCase):
"""测试city_functions.py"""
def test_first_last_name(self):
formatted_name=city_functions('santiago','chile')
self.assertEqual(formatted_name,'santiago,chile')
if __name__ == '__main__':
unittest.main()
导入
def city_functions(City, Country):
return f"{City},{Country}"
测试通过
修改上述函数,将形参population设置为可选的。再次运行test_cities.py,确认测试test_city_country()又通过了。
再编写一个名为test_city_country_population()的测试,核实可以使用类似于’santiago’、'chile’和’population=5000000’这样的值来调用这个函数。
再次运行test_cities.py,确认测试test_city_country_population()通过了。
修改后的函数
def city_functions(City, Country,population=0):
return f"{City},{Country}-population {population}"
主函数不变,测试不通过
import unittest
from city_functions import city_functions
class NamesTestCase(unittest.TestCase):
"""测试city_functions.py"""
def test_first_last_name(self):
formatted_name=city_functions('santiago','chile')
self.assertEqual(formatted_name,'santiago,chile')
if __name__ == '__main__':
unittest.main()
主函数修改后,测试通过
import unittest
from city_functions import city_functions
class NamesTestCase(unittest.TestCase):
"""测试city_functions.py"""
def test_first_last_name(self):
formatted_name=city_functions('santiago','chile')
self.assertEqual(formatted_name,'santiago,chile-population 0')
if __name__ == '__main__':
unittest.main()
编写一个名为give_raise()的方法,它默认将年薪增加5000美元,但也能够接受其他的年薪增加量。
为Employee编写一个测试用例,其中包含两个测试方法:
test_give_default_raise()和test_give_custom_raise()。
使用方法setUp(),以免在每个测试方法中都创建新的雇员实例。运行这个测试用例,确认两个测试都通过了。
主程序
import unittest
from Employee import Employee
class TestEmployee(unittest.TestCase):
def setUp(self):
self.employee=Employee('liu','yi zheng',100_000)
def test_give_default_raise(self):
self.assertIn('liu',self.employee.first_name)
def test_give_custom_raise(self):
self.assertIn(str(100_000),str(self.employee.years_money))
if __name__ == '__main__':
unittest.main()
函数
class Employee:
def __init__(self,first_name,last_name,years_money):
self.first_name=first_name
self.last_name=last_name
self.years_money=years_money
def give_raise(self,money_rise=5000):
self.years_money += money_rise
测试结果