11.1
city_functions.py
def get_city_country(City,Country): return City+", "+Country
test_cities.py
import unittest from city_functions import get_city_country class CityTestCase(unittest.TestCase): def test_city_country(self): city_country=get_city_country('Milan','Italy') self.assertEqual(city_country,'Milan, Italy') unittest.main()
11.2
city_functions.py
population必不可少时
def get_city_country(City,Country,population):
return City+", "+Country+' - '+population
population可选时
def get_city_country(City,Country,population=''): if population: return City+", "+Country+' - '+population else: return City+", "+Country
test_cities.py
import unittest from city_functions import get_city_country class CityTestCase(unittest.TestCase): def test_city_country(self): city_country=get_city_country('Milan','Italy') self.assertEqual(city_country,'Milan, Italy') def test_city_country_population(self): city_country=get_city_country('Milan','Italy','population=8000000') self.assertEqual(city_country,'Milan, Italy - population=8000000') unittest.main()
11.3
employee_class.py
class Employee: def __init__(self,firstname,lastname,annual_salary): self.firstname=firstname self.lastname=lastname self.annual_salary=annual_salary def give_raise(self,add=5000): self.annual_salary+=addtest_employees.py
import unittest from employee_class import Employee class EmplyeeTestCase(unittest.TestCase): def setUp(self): self.my_employee=Employee('liang','junan',100000) def test_give_default_raise(self): self.my_employee.give_raise() self.assertEqual(self.my_employee.annual_salary,105000) def test_give_custom_raise(self): self.my_employee.give_raise(6400) self.assertEqual(self.my_employee.annual_salary,106400) unittest.main()