《Python 编程-从入门到实践》11-1~11-3

11-1 城市和国家。

import unittest


def city_functions(city,country):
	return city.title()+","+country.title()


class A(unittest.TestCase):
	def setUp(self):
		self.case = city_functions("santiago","chile")
		
	def test_ill(self):
		self.assertEqual(self.case,"Santiago,Chile")
		print("Equal")
		
unittest.main()

11-2 人口数量：

import unittest

def city_functions(city,country,population=""):
	if(population):
		return city.title()+","+country.title()+" - population "+population
	else:
		return city.title()+","+country.title()

class A(unittest.TestCase):		
	def test_ill(self):
		self.assertEqual(city_functions("santiago","chile"),"Santiago,Chile")
		self.assertEqual(city_functions("santiago","chile","5000000"),"Santiago,Chile - population 5000000")
		print("Equal")
		
unittest.main()

11-3 雇员：

import unittest

class Employee():
	def __init__(self,lastname,firstname,salary):
		self.lastname = lastname
		self.firstname = firstname
		self.salary = salary
	
	def get_raise(self,rsalary=5000):
		self.salary += rsalary
		return self.salary
		

class A(unittest.TestCase):		
	def setUp(self):
		self.test1 = Employee('Hua','Li',10000).get_raise()
		self.test2 = Employee('Hua','Li',10000).get_raise(1000)
	
	def test_give_default_raise(self):
		self.assertEqual(self.test1,15000)
	
	def test_give_custom_raise(self):	
		self.assertEqual(self.test2,11000)
		
unittest.main()