第十一章 测试代码
11-1 城市和国家
import unittest
def func(city, country):
return city + ", " + country
class FunctionTest(unittest.TestCase):
def test_city_country(self):
fomatted = func("santiago", "chile")
self.assertEqual(fomatted, "santiago, chile")
unittest.main()
11-2 人口数量
import unittest
def func(city, country, population=1000000):
return city + ", " + country + ": " + str(population)
class FunctionTest(unittest.TestCase):
def test_city_country_population(self):
fomatted = func("santiago", "chile", population=1000)
self.assertEqual(fomatted, "santiago, chile: 1000")
unittest.main()
11-3 雇员
import unittest
class Employee():
def __init__(self, name, salary=10000):
self.name = name
self.salary = salary
def give_raise(self, increment=5000):
self.salary += increment
class TestEmployee(unittest.TestCase):
def setUp(self):
self.employee = Employee("Alice")
def test_give_default_raise(self):
self.employee.give_raise()
self.assertEqual(self.employee.salary, 15000)
def test_give_custom_raise(self):
self.employee.give_raise(6000)
self.assertEqual(self.employee.salary, 16000)
unittest.main()