单元测试与员工类实战-优快云博客

本文介绍了Python中单元测试的基本用法，通过两个具体案例进行说明：城市与国家信息格式化函数的测试及员工类中薪资调整方法的测试。文章详细展示了如何编写测试用例以确保函数和类的方法按预期工作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、第十一章作业。

11-1

city_functions.py:

def city_country(city, country, population):

cityCountry = city.title() + ", " + country.title()

return cityCountry

test_cities.py:

unittest.main()

11-2

city_functions.py:

def city_country(city, country, population):

cityCountry = city.title() + ", " + country.title() + " - " + str(population)

return cityCountry

city_functions.py:

def city_country(city, country, population = 'XXX'):
if population != 'XXX':
cityCountry = city.title() + ", " + country.title() + " - " + str(population)
else:
cityCountry = city.title() + ", " + country.title()

return cityCountry

test_cities.py:

import unittest
from city_functions import city_country
class testCase(unittest.TestCase):
"""docstring for test_city_country"""
def test_city_country(self):
testCityCountry = city_country('Santiago', 'Chile')
self.assertEqual(testCityCountry, 'Santiago, Chile')
def test_city_country_population(self):
testCityCountry = city_country('santiago', 'chile', 5000000)
self.assertEqual(testCityCountry, 'Santiago, Chile - population 5000000')

unittest.main()

11-3

Employee.py:

class Employee:
def __init__(self, first_name, last_name, bound = 1000):
self.first_name = first_name
self.last_name = last_name
self.bound = bound
def give_raise(self, buond_r = 5000):
self.bound += buond_r

return buond_r

test_employee.py:

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):
def setUp(self):
self.test1 = Employee("Xu", "Yi")
self.test2 = Employee('Zhu', 'Gaojie', 2000)

def test_give_default_raise(self):
self.test1 = self.test1.give_raise()
self.assertEqual(str(self.test1), '5000')
def test_give_custom_raise(self):
self.test2 = self.test2.give_raise(2000)
self.assertEqual(str(self.test2), '2000')