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+=add
test_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()