11-1 城市和国家
代码:
1、city_functions.py:
def show_city(city,country):
message = city + ", " + countryreturn message
2、test.py:
import unittest
from city_functions import show_city
class CityTestCase(unittest.TestCase):
def test(self):
answer = show_city('Santiago','Chile')
self.assertEqual(answer,'Santiago, Chile')
unittest.main()
运行结果:
11-2 人口数量
(1)代码:
city_functions.py:
def show_city(city,country,population):
message = city + ", " + country + " - population " + str(population)
return message
运行结果:
(2)代码:
city_functions.py:
def show_city(city,country,population=0):
if population==0:
message = city + ", " + country
else:
message = city + ", " + country + " - population " + str(population)
return message
运行结果:
(3)代码:
test.py:
import unittest
from city_functions import show_city
class CityTestCase(unittest.TestCase):
def test(self):
answer = show_city('Santiago','Chile')
self.assertEqual(answer,'Santiago, Chile')
def test_city_country_population(self):
answer = show_city('Santiago','Chile',population = 5000000)
self.assertEqual(answer,'Santiago, Chile - population=5000000')
unittest.main()
运行结果:
11-3 雇员
代码:
import unittest
from employee import Employee
class EmployeeTestCase(unittest.TestCase):
def setUp(self):
self.myemployee = Employee('Mike','Storm',10000)
def test_give_default_raise(self):
self.myemployee.give_raise()
self.assertEqual(self.myemployee.salary,15000)
def test_give_custom_raise(self):
self.myemployee.give_raise(2500)
self.assertEqual(self.myemployee.salary,12500)
unittest.main()
运行结果: