第11章 测试代码

10-11喜欢的数字

import json
num=input('Please input your favorite number!')
with open('number.json','w') as f_obj:
	json.dump(num,f_obj)
	
with open('number.json','r') as f_obj:
	new_num=json.load(f_obj)
	print("I know your favorite number,it's %s" % new_num)

11-1城市和国家

import unittest

def city_country(city,country):
	return city+', '+country
	
class TestCase(unittest.TestCase):
	def test_city_coutry(self):
		format=city_country('Santiago','Chile')
		self.assertEqual(format,'Santiago, Chile')
		
unittest.main()

11-2人口数量

import unittest

def city_country(city,country,population=0):
	if population:
		return city+', '+country+' - '+population
	else:
		return city+', '+country
class TestCase(unittest.TestCase):
	def test_city_coutry(self):
		format=city_country('Santiago','Chile','population=5000000')
		self.assertEqual(format,'Santiago, Chile - population=5000000')
		
unittest.main()

11-3雇员

import unittest

class Employee():
	def __init__(self,first_name,last_name,salary):
		self.first_name=first_name
		self.last_name=last_name
		self.salary=salary
		
	def give_raise(self):
		self.salary+=5000
		
class TestEmployee(unittest.TestCase):
	def setUp(self):
		self.my_survey=Employee('Lily','Green',10000)
		self.first_name='Lily'
		self.last_name='Green'
		self.salary=10000
		
	def test_give_default_raise(self):
		self.assertEqual(self.first_name,self.my_survey.first_name)
		self.assertEqual(self.first_name,self.my_survey.first_name)
		self.assertEqual(self.first_name,self.my_survey.first_name)
		
	def test_give_custom_raise(self):
		self.my_survey.give_raise()
		self.assertEqual(self.first_name,self.my_survey.first_name)
		self.assertEqual(self.first_name,self.my_survey.first_name)
		self.assertEqual(self.first_name,self.my_survey.first_name)
		
unittest.main()